在 rapids.ai docker 容器中持久安装 pip

Question

这可能是一个非常愚蠢的问题，但必须从某个地方开始。我正在玩 NVDIA 的 rapids.ai gpu-enhanced docker 容器，但这（大概是设计使然）并没有与 pytorch 一起提供。现在，当然，我每次都可以做一个 pip install torch torch-ignite，但这既烦人又耗费资源（而且 pytorch 是一个很大的下载量）。在容器中持久化 pip install 的批准方法是什么？

Answer 1

创建一个新的 Dockerfile 以现有图像为基础构建新图像：

FROM the/rapids-ai/image

RUN pip install torch torch-ignite

然后

$ ls Dockerfile
Dockerfile
$ docker build -t myimage .

您现在可以：

$ docker run myimage

在 rapids.ai docker 容器中持久安装 pip

persistent pip install in rapids.ai docker container

pip

docker

pytorch

rapids