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

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

0 голосов
/ 27 ноября 2018

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

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

ОС: Ubuntu 18.04

Nvidia RTX 2080 Ti * 2

Nvidia driver-415

CUDA: 10,0

cuDNN: 7,3,0 тензор потока: 1.11,0 

import tensorflow

Ошибка:

Traceback (последний вызов был последним): Файл "/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_moduleload_dynamic (имя, имя файла, файл) Файл "/usr/lib/python3.6/imp.py", line 343, в load_dynamic return _load (spec) ImportError: libcublas.so.9.0: невозможно открыть общий объектный файл: нет такого файла или каталога

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

Traceback (последний вызов был последним): файл "", строка 1, в файле "/usr/local/lib/python3.6/dist-packages/tensorflow/init.py", строка 22, из файла tenorflow.python import pywrap_tensorflow # pylint: disable = unused-import File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/init.py", строка 49, из файла tenorflow.python импортировать файл pywrap_tensorflow "/usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow.py", строка 74, в повышение ImportError (msg) ImportError: Traceback (последний вызовпоследний): файл "/usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow.py", строка 58, из файла tenorflow.python.pywrap_tensorflow_internal import * File "/ usr / local / lib /python3.6 / расстояние-пакеты / tensorflow / питон / 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_('_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: невозможно открыть общий объектный файл: нет такого файла или каталога

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

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

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

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

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

Я уже проложил путь к приложению в ~ / .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

1 Ответ

0 голосов
/ 27 ноября 2018

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

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

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

...