ImportError: при попытке запустить TensorFlow Image с поддержкой графического процессора имя модуля не отображается

Я пытаюсь установить тензор потока в моем ноутбуке с графическим процессором NVIDIA 1050 Ti. Я следовал за шагами в:

https://www.tensorflow.org/install/docker

Под поддержкой GPU я использовал lspci | grep -i nvidia для проверки доступности графического процессора. Затем я сделал следующее, чтобы проверить установку nvidia-docker:

docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi

После выполнения появляется таблица с заголовком NVIDIA-SMI 410.48 и версией драйвера: 410.48.

Я сделал: pip3 install tensorflow-gpu уже, но когда я выполнил (я заменил python на python3):

docker run --runtime=nvidia -it --rm tensorflow/tensorflow:latest-gpu python3 -c "import tensorflow as tf; tf.enable_eager_execution(); print(tf.reduce_sum(tf.random_normal([1000, 1000])))"

Я получил:

Traceback (most recent call last):
 File "<string>", line 1, in <module>
ImportError: No module name 'tensorflow'

Не могли бы вы помочь решить эту проблему?

0 ответов

Другие вопросы по тегам