在具有 GPU 支持的 Tensorflow 的 Docker 映像中(例如:tensorflow/tensorflow:2.2.0-gpu
)安装的python包是tensorflow-gpu
(如图所示pip freeze
).
安装任何依赖于的 python 包tensorflow
触发tensorflow本身的安装,尽管它已经以不同的名称安装了(因为——正确的是——tensorflow-gpu
!= tensorflow
).
有办法避免这种情况吗?
您可以添加安装假冒的指令tensorflow
“package”只写入元数据而不添加重复的源:
$ 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
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)