Tensorflow：docker鏡像和-gpu后綴

Question

In the Docker image for Tensorflow with GPU support (for example: tensorflow/tensorflow:2.2.0-gpu ) the installed python package is tensorflow-gpu (as shown in pip freeze ).

Installing any python package that depends on tensorflow triggers the installation of tensorflow itself, although it's already installed under a different name (because -- correctly -- tensorflow-gpu != tensorflow ).

有沒有辦法避免這種情況？

Answer 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