前提:成功安装显卡驱动和CUDA
1.安装docker
curl -fsSL https://get.docker.com | bash -s docker --mirror Aliyun # 从阿里云下载docker
sudo groupadd docker # 添加docker group
sudo gpasswd -a ${USER} docker # 将当前用户添加到docker组
sudo service docker restart # 重启docker
newgrp docker # 更新用户组
关闭窗口重新打开终端,输入测试指令docker version验证docker是否成功安装好。
2.安装NVIDIA-Docker2
清理之前的安装版本
sudo docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f
sudo apt-get purge -y nvidia-docker
sudo apt autoremove
输入如下指令:
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
###
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 pkill -SIGHUP dockerd
安装完输入如下指令进行测试:
sudo docker run --rm --gpus all nvidia/cuda:11.1.1-base nvidia-smi
11.1.1-base可自行选择,版本号可参考下边这个网址:
doc/supported-tags.md · master · nvidia / container-images / cuda · GitLab
如输出如下界面则说明NVIDIA-Docker2安装成功。
主要参考这几位博主:
https://blog.csdn.net/quantum7/article/details/86416600/?ops_request_misc=&request_id=&biz_id=102&utm_term=ubuntu%20%20nvidia-docker2%E5%AE%89%E8%A3%85&utm_medium=distribute.pc_search_result.none-task-blog-2~all~sobaiduweb~default-0-86416600.142^v93^chatgptT3_2&spm=1018.2226.3001.4187
https://blog.csdn.net/m0_61070051/article/details/127746310
Ubuntu18.04安装nvidia-docker2_unable to find image 'nvidia/cuda:latest' locally_城俊BLOG的博客-CSDN博客