在具有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] 删除。
我来说两句