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

尼古拉(Nicola Montecchio)

在具有GPU支持的Tensorflow的Docker映像中(例如:),tensorflow/tensorflow:2.2.0-gpu已安装的python软件包为tensorflow-gpu(如所示pip freeze)。

安装任何依赖的python软件包都会tensorflow触发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

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章