Tensorflow:docker图像和-gpu后缀

在具有GPU支持的Tensorflow的Docker映像中（例如：

Tensorflow/Tensorflow:2.2.0-GPU

），安装的python包是

Tensorflow GPU

（如

pip冻结

所示）

安装依赖于

tensorflow

的任何python包都会触发tensorflow本身的安装，尽管它已经以不同的名称安装（因为--正确地--

tensorflow gpu

！=

tensorflow

）

---

有什么方法可以避免这种情况吗？

您可以添加一条指令来安装一个假的

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…

一行程序是一件相当艺术的作品：）谢谢，效果很好：-）很高兴我能帮上忙！