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

Question

这可能是一个非常愚蠢的问题，但总得有个开始吧。我正在玩NVDIA的rapids.ai图形处理器增强型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