是否可以同时安装张力stream的CPU和GPU版本

我正在使用nvidia-docker从Docker容器中访问GPU。 但是，并不是所有的机器都具有GPU，我想在GPU不可用时自动回退到CPU版本。

我是否必须构build独立的Docker镜像 – 一个用于CPU，另一个用于GPU – 或者可以安装tensorflow和tensorflow-gpu并根据GPU是否可用select正确的版本？

Answer 1

您可能需要查看官方的Tensorflow docker 图片 GPU版本使用nvidia-docker docker来访问GPU。

我过去所做的是有两个几乎相同的Dockefiles。 Dockerfiles中的唯一区别是FROM指令：

 FROM tensorflow/tensorflow:latest-py3 

要么

 FROM tensorflow/tensorflow:latest-gpu-py3 

（如果你愿意的话，你也可以selectPython2的图片）

其他的都是一样的，你甚至可以自动执行这个操作，这样当你构build图像时就会设置合适的FROM标签。 我已经使用makefile来build立适当的图像取决于主机是否有GPU。