Как установить 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