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

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