Tag:

在Docker容器或Docker镜像中使用Tensorflow的正确方法是什么?

我试图运行一个简单的docker容器与Tensorflow可用(首先与CPU)。 我认为这将是一个好主意,只设置我的Dockerimage只有一次(即不更新张量stream版本每次我运行一个容器)。 为了做到这一点,我build议在我的Dockerfile中做如下操作(注释源自给出了我的build议): # This means you derive your docker image from the tensorflow docker image FROM gcr.io/tensorflow/tensorflow 然而,当我运行我的Docker容器时,我做了pip list ,并没有看到任何地方Tensorflow可用,当我跑我的脚本我得到了熟悉的错误: ImportError: No module named 'tensorflow' 我想通过只有我的Dockerfile明确pip3 install tensorflow解决这个问题的方法。 我打算做一个bash脚本,让我的Dockerfile运行它: # bash script intall_tensorflow.sh # to install Tensorflow in container export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-0.12.0rc1-py3-none-any.whl pip3 install –upgrade $TF_BINARY_URL 然后添加到docker文件中: RUN sh intall_tensorflow.sh 然而,我的直觉告诉我这可能是错的或太hacky。 为什么我要在第一个地方需要tensorflow base image FROM […]