docker就是一个系统,使用docker配置环境比较方便
1.拉镜像
docker pull pytorch/pytorch:latest
此处建议使用阿里云加速
https://yq.aliyun.com/articles/29941
2.查看本地有的镜像
docker images
3.查看本地运行的docker
docker ps
4.运行一个镜像
docker run -d -it -v path1:path2 --shm-size "100G" --runtime=nvidia pytorch/pytorch:latest /bin/bash
-d代表后台,-it代表交互式,-v用来挂载磁盘到镜像里面,--shm-size设置镜像内存大小,否则在训练模型的时候容易出现shm
5.进入一个镜像
docker exec -it 进程ID /bin/bash