Docker中构建深度学习环境

在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