Как установить TensorFlow-gpu с cuda8.0?

Я попытался установить его в соответствии с инструкциями на официальном веб-сайте, что приводит к ImportError при импорте shadoworflow:

ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

Я запускаю код cat/usr/local/cuda/version.txt, который показывает, что моя версия cuda - 8.0.61.

Кажется, что tensorflow ищет cuda 9.0. Я не могу обновить cuda, поскольку я работаю с общим сервером gpu, и у меня нет полномочий root.

Есть ли способ сделать shadoworflow работать с cuda 8.0? Или любой другой способ?

Спасибо!!

Ответ 1

Я сталкивался с подобной проблемой, пока не нашел

https://www.tensorflow.org/install/install_sources#tested_source_configurations

проверьте установленную версию cuda и версию cudnn, а затем выясните, какая версия tennsflow-gpu совместима с версиями, указанными выше.

Я установил cuda 8 и cudnn v5.1, поэтому, проверив вышеуказанную ссылку, тензор-gpu 1.2.0 был совместим и после его установки использовал

pip install tensorflow-gpu==1.2.0

Это сработало для меня.

Ответ 2

Вам нужно будет установить версию 1.4.1 для CUDA-8 как

pip install tensorflow-gpu==1.4.1

Последняя версия (версия 1.5) предназначена для CUDA-9