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"