需要安装可以正常使用 gpu 的 docker
只是为了自己记录,方便以后使用,可以去原文处看看,参考链接,优秀
卸载旧版本 docker (如果有)
sudo apt remove docker docker-engine docker-ce docker.io
安装 docker
sudo apt-get install apt-transport-https ca-certificates curl gnupg-agent software-properties-common
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
sudo apt-key fingerprint 0EBFCD88
sudo add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable "
sudo apt update
sudo apt-get install docker-ce docker-ce-cli containerd.io
安装 toolkit
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 update
sudo apt install -y nvidia-container-toolkit
重启 docker 服务
sudo service docker restart
查看 docker 版本
sudo docker version
测试 gups 参数是否可用
docker run --help | grep -i gpus
免 sudo 使用 docker
- 不是必须的,但是很方便
sudo groupadd docker
sudo gpasswd -a ${USER} docker
sudo service docker restart
newgrp - docker