Назад | Перейти на главную страницу

Установите драйверы Nvidia 9.0 для пакета TensorFlow (Debian 9.7)

Я установил драйверы Nvidia 9.1 на свой Debian 9.7 (Dataproc), когда я пытаюсь запустить TensorFlow 1.9 через этот тестовый скрипт, он терпит неудачу:

Использовал это руководство для установки драйверов графического процессора: https://cloud.google.com/dataproc/docs/concepts/compute/gpus

Используемый pip install tensorflow-gpu

import tensorflow as tf

Я получил:

>>> import tensorflow as tf
Traceback (most recent call last):
  File "/opt/tony/TonY-samples/deps/tf-gpu/lib/python3.5/site-packages/tensorflow/python/pywrap_tensorflow.py", line 58, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "/opt/tony/TonY-samples/deps/tf-gpu/lib/python3.5/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "/opt/tony/TonY-samples/deps/tf-gpu/lib/python3.5/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper
    _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
  File "/opt/tony/TonY-samples/deps/tf-gpu/lib/python3.5/imp.py", line 242, in load_module
    return load_dynamic(name, filename, file)
  File "/opt/tony/TonY-samples/deps/tf-gpu/lib/python3.5/imp.py", line 342, in load_dynamic
    return _load(spec)
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

Версия CUDA

cat /usr/lib/cuda/version.txt 
CUDA Version 9.1.85

TensorFlow (Пробовал с 1.9 и 1.12)

pip freeze
absl-py==0.7.0
astor==0.7.1
gast==0.2.2
grpcio==1.18.0
h5py==2.9.0
Keras-Applications==1.0.7
Keras-Preprocessing==1.0.9
Markdown==3.0.1
numpy==1.16.1
protobuf==3.6.1
six==1.12.0
tensorboard==1.12.2
tensorflow-gpu==1.12.0
termcolor==1.1.0
Werkzeug==0.14.1

Версия Debian

cat /etc/debian_version
9.7

Я установил драйверы, следуя этому руководству:

Следование этому руководству:

Как установить CUDA 9.0 CUDNN 7.0 автоматически без подсказок? У apt-get нет 9.0

./cuda-linux.9.0.176-22781540.run --noprompt