一、环境配置
1.1 新建docker
sudo docker run --gpus all --privileged -it --shm-size 8G -v /home/user/xxxx:/workspace/ --name docker_name -p 8011:22 pytorch/pytorch:1.6.0-cuda10.1-cudnn7-devel
/home/user/xxxx
是你要映射到容器里面的文件夹(这个文件夹会映射到docker为我们自动创建的/workspace文件夹),--name
是容器的名字,后面-p 8011:22
是把docker内部的22端口映射到8011端口,pytorch/pytorch:1.6.0-cuda10.1-cudnn7-devel
是要使用的docker镜像
1.2 配置docker
- 使用命令
sudo docker start docker_name
sudo docker attach docker_name
启动并进入docker