Question

在具有GPU支持的Tensorflow的Docker映像中（例如：// some other file import { path } from './deps.ts' // instead of "https://deno.land/std@0.51.0/path/mod.ts" path.basename('/bar/test.txt');），已安装的python软件包为tensorflow/tensorflow:2.2.0-gpu（如tensorflow-gpu所示）。

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

有办法避免这种情况吗？

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

Tensorflow：泊坞窗映像和-gpu后缀

1 个答案: