使用Docker配置Pytorch环境

配置Docker环境首先要拉去镜像，可以利用生成的镜像直接拉取，也可自定义配置Dockerfile文件，生成镜像。

 

1. 直接拉取pytorch镜像

可以在Docker官网查找对应的pytorch镜像进行拉取。

如拉取最新版本的镜像

docker pull pytorch/pytorch:latest

 

 2.  配置Dockerfile文件

Pytorch Docker镜像构建教程（不同系统、CUDA、Python版本） - 知乎 (zhihu.com)

尽管Pytorch官方和Nvidia-NCG提供了很多版本的Pytorch镜像，但是我们平时的科研和工作中经常需要在特定系统版本、特定CUDA版本、特定Python版本下使用特定版本的Pytorch，官方的镜像很难满足自定义的需求，因此需要自己构建Pytorch镜像。下面我们来一步一步完成Pytorch镜像的构建。