使用 Pytorch-cuda 作为 Docker 的基础镜像

Question

我需要使用带有 cuda gpu 访问权限的 Pytorch。我使用 FROM "pytorch/pytorch:1.9.0-cuda10.2-cudnn7-runtime" 作为我的基础镜像。这个命令是否也安装了必要的python版本？

Answer 1

从 DockerHub 您可以看到这些类型的镜像是如何构建的： https://hub.docker.com/layers/pytorch/pytorch/1.9.0-cuda10.2-cudnn7-runtime/images/sha256-5dc11a9036bcb5b7950f4f8a43974057559278fa4bd87513cafd477fef9eb57a?context=explore

在那里您可以看到 conda 已安装。我建议在您正在使用的 DockerFile 中使用它附带的 conda。