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

标签 docker pip pytorch rapids

这可能是一个非常愚蠢的问题,但是必须从某个地方开始。我正在使用NVDIA的Rapids.ai gpu增强的docker容器,但这(大概是设计使然)不带有pytorch。现在,当然,我每次都可以做一个pip install torch torch-ignite,但这既烦人又消耗资源(而pytorch的下载量很大)。在容器中保留pip install的批准方法是什么?

最佳答案

创建一个新的Dockerfile,以根据现有的图像生成一个新图像:

FROM the/rapids-ai/image

RUN pip install torch torch-ignite

然后

$ ls Dockerfile
Dockerfile
$ docker build -t myimage .

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

$ docker run myimage

关于docker - 在Rapids.ai Docker容器中持久安装pip,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/61463624/

相关文章:

int - 如何在 PyTorch 中将 boolean 张量转换为整数?

python - 如何解决安装django-simple-history的问题?

python - 使用beam_search(huggingface库)生成文本时出现不匹配的张量大小错误

docker - 从外部容器向 docker 上的 statsd/ Graphite 数据库发送数据时出现问题

bash - RUN 命令中字符串中的 Dockerfile ARG 替换

python - 错误 "ModuleNotFoundError: No module named ' numpy'";但已安装 NumPy

python - 如何在服务器上没有 root 访问权限的情况下完全安装 numpy?

pytorch - 无法将 PyTorch 模型导出到 ONNX

spring-boot - 第一次调用时出现 ZuulException (SendErrorFilter)

php - 如何在docker中使用unix套接字将nginx连接到php-fpm