постоянный пункт установки в Rapids.ai docker контейнер

Question

Это, наверное, действительно глупый вопрос, но с чего-то начинать нужно. Я играю с контейнером NVIDIA rapids.ai gpu 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