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'
Не могли бы вы помочь решить эту проблему?