Tensorflow：docker 镜像和 -gpu 后缀

【问题标题】：Tensorflow: docker image and -gpu suffixTensorflow：docker 镜像和 -gpu 后缀
【发布时间】：2020-09-07 02:11:33
【问题描述】：

在支持 GPU 的 Tensorflow 的 Docker 镜像中（例如：tensorflow/tensorflow:2.2.0-gpu），安装的 python 包是tensorflow-gpu（如pip freeze 所示）。

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

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

【问题讨论】：

标签： docker tensorflow pip dependencies gpu

【解决方案1】：

您可以添加一条指令来安装一个虚假的tensorflow“包”，它只写入元数据而不添加重复的源：

$ 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 包，但 -c ... 单线是一件艺术品 :) 谢谢，效果很好
:-) 很高兴我能帮上忙！