在Docker中构建深度学习环境
参考博客:https://blog.csdn.net/ouening/article/details/105932177
1.安装nvidia-docker
$ distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
$ curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
$ curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
$ sudo apt-get update && sudo apt-get install -y nvidia-docker2
$ sudo systemctl restart docker
2.拉去所需pytorch docker镜像
**镜像的地址列表:**https://hub.docker.com/r/pytorch/pytorch/tags?page=1
比如我拉取的是:pytorch/pytorch:1.3-cuda10.1-cudnn7-devel,则执行
$ sudo docker pull pytorch/pytorch:1.3-cuda10.1-cudnn7-devel
查看安装的镜像:
$ sudo docker images
如下图所示 pytorch/pytorch:1.3-cuda10.1-cudnn7-devel就是我们拉取的镜像
3.启动容器
$ sudo docker run --name FastReID --gpus all -it -v $PWD:/home pytorch/pytorch:1.3-cuda10.1-cudnn7-devel bash
run
:代表启动一个容器,还有其他命令,例如rm
(删除容器),exec
,stop
,kill
等,可以查看链接:https://www.runoob.com/docker/docker-command-manual.html;--name
:代表容器名称,自行确定,如果没有指定,会随机生成;--gpus
:指定使用的gpu数量,这里为所有all
,一定要设置该参数,否则无法调用gpu;-it
:使用交互式interactive;-v
:磁盘挂载,$PWD:/home
的意思是将当前路径挂载到容器/home
目录下,这样进入容器之后cd
到/home
,就可以运行当前路径下的文件了;- pytorch/pytorch:1.3-cuda10.1-cudnn7-devel:表示docker镜像
- -
bash
:最后一项表示执行容器的命令,在交互模式下执行bash,就可以进入容器的bash了
4.停止容器
$ sudo docker stop FastReID
FastReID就是之前创建容器时赋予的名字
5.查看容器
$ sudo docker ps
6.也可以使用下面的命令(使用容易的ID)停止容器
$ sudo docker stop a25296453d73
7.查看所有容器,包括已经停止的容器
$ sudo docker ps -a
8.删除容器
一定要慎重,不然在容器中安装的环境就没了,又得重新安装
$ sudo docker rm FastReID
$ sudo docker rm a25296453d73
9.删除镜像
$ sudo docker image rm ${IMAGE ID}
可以删除镜像,如果提示有容器在占用,需要先删除容器,再删除镜像,这里的${IMAGE ID}
就是镜像的ID