Tensorflow: docker 图片和-gpu 后缀

Question

在支持GPU的Tensorflow的Docker镜像中（例如：tensorflow/tensorflow:2.2.0-gpu）安装的python包是tensorflow-gpu（如[=13=所示） ]).

安装依赖于 tensorflow 的任何 python 包会触发 tensorflow 本身的安装，尽管它已经以不同的名称安装（因为 -- 正确地 -- tensorflow-gpu != tensorflow)。

有没有办法避免这种情况？

Answer 1

您可以添加一条指令来安装仅写入元数据而不添加重复源的伪造 tensorflow "package"：

$ python -c 'from setuptools import setup; setup(name="tensorflow", version="2.2.0")' install

在 docker 图像中，它看起来像这样：

FROM tensorflow/tensorflow:2.2.0-gpu
RUN python -c 'from setuptools import setup; setup(name="tensorflow", version="2.2.0")' install
RUN pip install my-requirements
RUN pip uninstall -y tensorflow  # cleaning up

Tensorflow: docker 图片和-gpu 后缀

Tensorflow: docker image and -gpu suffix

dependencies

gpu

pip

docker

tensorflow