【问题标题】:using tensorflow inside docker with gpu在带有gpu的docker内部使用tensorflow
【发布时间】: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


    【解决方案1】:

    “devel”图片包含张量流源,见:https://hub.docker.com/r/tensorflow/tensorflow/

    引用:

    开发图像随 Bazel 一起提供,非常适合开发更改 TensorFlow 大师。 /tensorflow_src 包含 TensorFlow 源代码 构建 Pip 包的最新每晚提交的树 成功在容器中。我们不再提供图片 在旧版本的 TF 之上开发(1.12.0 是最后一个版本 就是这种情况)。

    【讨论】:

    • 好的,我明白了。我只是没有想到 devel 分支只有源代码,并且也没有包含编译版本。谢谢。
    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 2018-02-11
    • 1970-01-01
    • 2018-02-15
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2021-10-21
    相关资源
    最近更新 更多