这可能是一个非常愚蠢的问题,但是必须从某个地方开始。我正在使用NVDIA的Rapids.ai gpu增强的docker容器,但这(大概是设计使然)不带有pytorch。现在,当然,我每次都可以做一个pip install torch torch-ignite
,但这既烦人又消耗资源(而pytorch的下载量很大)。在容器中保留pip install
的批准方法是什么?
最佳答案
创建一个新的Dockerfile
,以根据现有的图像生成一个新图像:
FROM the/rapids-ai/image
RUN pip install torch torch-ignite
然后
$ ls Dockerfile
Dockerfile
$ docker build -t myimage .
您现在可以执行以下操作:
$ docker run myimage
关于docker - 在Rapids.ai Docker容器中持久安装pip,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/61463624/