如何基于现有映像创建一个新的docker映像，但包含更多的Python包？

Question

假设我已经将NVIDIA NGC PyTorch docker图像拉到如下位置:docker pull nvcr.io/NVIDIA/PyTorch:21.07-py3

然后我想添加这些python包:Omegaconf、wandb、PycocoTools？

此外，如何在整个组织中分发新映像？

Answer 1

创建一个名为DockerFile的文件。添加下面解释的行。

添加from行以指定基本图像：

FROM nvcr.io/nvidia/pytorch:21.07-py3

将Pip升级到最新版本：

RUN python -m pip install --upgrade pip

安装您需要的附加Python包：

RUN python -m pip install omegaconf wandb pycocotools

总而言之，Dockerfile如下所示：

FROM nvcr.io/nvidia/pytorch:21.07-py3
RUN python -m pip install --upgrade pip
RUN python -m pip install omegaconf wandb pycocotools

在与dockerfile相同的目录中，运行此命令构建新映像，用您选择的名称替换my-new-image:

docker build -t my-new-image .

$ docker images
REPOSITORY                         TAG                            IMAGE ID       CREATED              SIZE
my-new-image                       latest                         082f76972805   13 seconds ago   15.1GB
nvcr.io/nvidia/pytorch             21.07-py3                      7beec3ff8d35   5 weeks ago          15GB
[...]

您现在可以运行新映像。

$ docker run --gpus all -it --rm --ipc=host my-new-image

..并验证它是否有附加的Python包：

# python -m pip list | grep 'omegaconf\|wandb\|pycocotools'
omegaconf                     2.1.1
pycocotools                   2.0+nv0.5.1
wandb                         0.12.1

Docker Hub存储库文档详细说明了以下步骤：

null

RUN useradd -ms /bin/bash myuser
USER myuser
ENV PATH "$PATH:/home/myuser/.local/bin"

由于这些原因，我发现让容器作为root运行是最容易的。