开发者社区

文档建议反馈控制台

文章/答案/技术大牛

发布

社区首页 >问答首页 >使用Pytorch-cuda作为Docker的基础镜像

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

Stack Overflow用户

提问于 2021-07-30 12:41:04

回答 1查看 229关注 0票数 0

我需要使用具有cuda gpu访问权限的Pytorch。我使用"pytorch/pytorch:1.9.0-cuda10.2-cudnn7-runtime“中的作为我的基础图像。必要的python版本是否也随此命令一起安装？

EN

回答 1

Stack Overflow用户

发布于 2021-07-30 15:22:50

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

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

票数 0

EN

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/68591135

复制

相似问题