Ошибка импорта tenorflow-gpu в ubuntu18.04

Question

Я получаю сообщение об ошибке при импорте tenorflow.

Технические характеристики моего компьютера:

ОС: Ubuntu 18.04

Nvidia RTX 2080 Ti * 2

Nvidia driver-415

CUDA: 10,0

cuDNN: 7.3.0 тензор потока: 1.11.0

import tensorflow

Ошибка:

Трассировка (последний вызов был последним): файл "/usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow.py", строка 58, из файла tenorsflow.python.pywrap_tensorflow_internal import * File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow_internal.py", строка 28, в файле _pywrap_tensorflow_internal = swig_import_helper()" /usr/local/lib/python3.6/dist-packages/tensorflow/ python / pywrap_tensorflow_internal.py ", строка 24, в swig_import_helper _mod = imp.load_module('_pywrap_tensorflow_internal', fp, путь, описание) Файл"/usr/lib/python3.6/imp.py", строка 243, в возвращении load_module load_dynamic(имя, имя файла, файл) Файл "/usr/lib/python3.6/imp.py", строка 343, в load_dynamic return _load(spec) ImportError: libcublas.so.9.0: невозможно открыть файл общего объекта: нет такого файл или каталог

Во время обработки вышеупомянутого исключения произошло другое исключение:

Traceback (последний вызов был последним): файл "", строка 1, в файле "/usr/local/lib/python3.6/dist-packages/tensorflow/init.py", строка 22, из тензорного потока.python import pywrap_tensorflow # pylint: disable = unused-import File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/init.py", строка 49, из файла tenorflow.python import из файла pywrap_tensorflow "/ usr / local /lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow.py ", строка 74, в повышение. ImportError(msg) ImportError: трассировка (последний вызов был последним): файл" / usr / local / lib / python3. 6 / dist-packages / tenorflow / python / pywrap_tensorflow.py ", строка 58, из файла tenorflow.python.pywrap_tensorflow_internal import * Файл" /usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow_internal. py ", строка 28, в файле _pywrap_tensorflow_internal = swig_import_helper()" /usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow_internal.py ", строка 24, в swig_import_helper _mod = imp.load_module (_pywrap_tensorfl ow_internal ', fp, pathname, description) Файл "/usr/lib/python3.6/imp.py", строка 243, в load_module возвращает файл load_dynamic(имя, имя файла, файл) Файл "/usr/lib/python3.6/ imp.py ", строка 343, в load_dynamic return _load(spec) ImportError: libcublas.so.9.0: невозможно открыть общий объектный файл: такого файла или каталога нет

Не удалось загрузить собственную среду выполнения TensorFlow.

См. https://www.tensorflow.org/install/install_sources.

по некоторым распространенным причинам и решениям. При обращении за помощью включите всю трассировку стека над этим сообщением об ошибке.

Я уже пытался использовать Ubuntu 16.04, но GPU не поддерживал. Установка CUDA9.0 и CUDA9.2 тоже не поддерживалась.

Как я могу использовать tenorflow-GPU?

Я уже app путь в ~/.bashrc

export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda-10.0
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64$LD_LIBRARY_PATH

Answer 1

tensorflow-gpu пакет построен на Cuda 9.0, но у вас установлена ​​Cuda 10.0.

Вам нужно либо понизить версию Cuda до 9.0 (но, если я помню, это невозможно с 2080Ti), либо создать тензорный поток из исходников. Существует обширная документация о том, как это сделать, на веб- сайте tenorflow

Вы также можете попробовать установить пакет tf-nightly-gpu, Следует отметить, что эта версия более экспериментальная, так как она не была протестирована так широко.