从无到有实现docker上部署GPU版深度学习环境

安装docker

Linux下安装

• 以ubuntu为例

方式

XShell下

• 依次执行命令

$ curl -fsSL https://get.docker.com -o get-docker.sh
$ sudo sh get-docker.sh

手动安装

• 依次执行以下命令

sudo apt-get update
sudo apt-get install \
    apt-transport-https \
    ca-certificates \
    curl \
    software-properties-common
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
sudo add-apt-repository \
   "deb [arch=amd64] https://download.docker.com/linux/ubuntu \
   $(lsb_release -cs) \
   stable"
sudo apt-get update
sudo apt-get install docker-ce
sudo docker run hello-world

• 出现如下信息，表示安装成功
  

启动关闭

• 安装完之后默认启动了docker，

# 启动docker
sudo service docker start

# 停止docker
sudo service docker stop

# 重启docker
sudo service docker restart

Windows下安装

• 参考https://www.runoob.com/docker/windows-docker-install.html

疑难杂症

• 大概率会遇到以下警告或者说错误

Got permission denied while trying to connect to the Docker daemon socket at unix:///var/run/docker.sock: Get http://%2Fvar%2Frun%2Fdocker.sock/v1.24/version: dial unix /var/run/docker.sock: connect: permission denied

执行命令：

sudo gpasswd -a $USER docker
newgrp docker

安装NVIDIA驱动

• 参考我的另一篇博客：传送门

安装nvidia-docker

• 参考我的另一篇博客：传送门

部署镜像

• 可以从dockerhub拉取镜像，注意使用gpu来部署，这样torch.cuda.is_available()才是True

docker pull pytorch/pytorch:1.8.1-cuda11.1-cudnn8-runtime