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

Question

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

Answer 1

创建一个Dockerfile基于现有图像构建新图像的新图像：

FROM the/rapids-ai/image

RUN pip install torch torch-ignite

接着

$ ls Dockerfile
Dockerfile
$ docker build -t myimage .

您现在可以执行以下操作：

$ docker run myimage