docker构建pytorch

配置

拉取镜像

（devel版本有nvcc和cudnn）
sudo docker pull pytorch/pytorch:1.6.0-cuda10.1-cudnn7-devel

构建容器

sudo docker run -it --gpus all --name lipytorch1.6 --shm-size="15g" -p 8181:8888 -v /data/docker/container/lipytorch1.6:/workspace pytorch/pytorch:1.6.0-cuda10.1-cudnn7-devel

维护

查看已经创建的容器

docker ps -a

查看运行中的容器

docker ps

启动容器

docker start "container id or name"

进入容器

docker attach "container id or name"

退出容器

exit

停止容器

docker stop "container id or name"

删除容器

docker rm "container id or name"

参考资料

https://blog.csdn.net/moutain9426/article/details/120283693