【发布时间】:2020-01-27 23:21:09
【问题描述】:
我正在尝试在 docker 中使用 tensorflow,按照以下说明进行操作
https://www.tensorflow.org/install/docker
我在 centos 7.7.1908 机器上安装了 docker 19.03,可以运行 nvidia-smi
docker run --gpus all nvidia/cuda:10.1-base nvidia-smi
我的理解是不推荐使用 nvidia-docker。然后我下载张量流
docker pull tensorflow/tensorflow:latest-devel-py3
或
docker pull tensorflow/tensorflow:devel-gpu
但是当我运行其中一个时
docker run -it --rm tensorflow/tensorflow:latest-devel-py3 python -c "import tensorflow as tf;"
我明白了
Traceback(最近一次调用最后一次):文件“”,第 1 行,in ModuleNotFoundError:没有名为“tensorflow”的模块未能 调整 tty 的大小,使用默认大小
我错过了什么?
好吧,后来我发现了
latest-gpu-py3
允许我导入 tensorflow,而
latest-devel-py3
没有。我原以为所有这些图像都可以使用...
【问题讨论】:
标签: docker tensorflow