☰

尝试运行支持 GPU 的 TensorFlow Image 时出现 ImportError：没有模块名称“tensorflow”

tag-icon • tag-icon nvidia gpu docker
尝试运行支持 GPU 的 TensorFlow Image 时出现 ImportError：没有模块名称“tensorflow”

我正在尝试在配备 NVIDIA 1050 Ti GPU 的笔记本电脑上安装 tensorflow。我遵循了以下步骤：

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

在 GPU 支持下，我使用 lspci | grep -i nvidia 来验证 GPU 是否可用。然后，我执行以下操作来验证 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'

您能帮忙解决这个问题吗？

相关内容