因为考虑到总是用pytorch的镜像,所以考虑直接拉一个pytorch的镜像,懒得直接安装

  1. 第一步进入docke hub页面 https://hub.docker.com/,直接搜索pytorch,然后选择如下图的pytorch官方的镜像:

  2. 然后选择自己需要的版本,我选择的是1.6的版本,然后选好
    在这里插入图片描述

  3. 最后点击这里复制,得到
    docker pull pytorch/pytorch:1.6.0-cuda10.1-cudnn7-devel
    然后直接输入命令行,可以得到镜像了:
    在这里插入图片描述

Logo

权威|前沿|技术|干货|国内首个API全生命周期开发者社区

更多推荐