一、更新显卡驱动到410.78版本
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-410
nvidia-smi
二、docker安装
1.卸载旧版本docker
sudo apt-get remove docker docker-engine docker.io
2.添加CA证书
sudo apt-get update
sudo apt-get install \
apt-transport-https \
ca-certificates \
curl \
software-properties-common
3.添加
curl -fsSL https://mirrors.ustc.edu.cn/docker-ce/linux/ubuntu/gpg | sudo apt-key add -
4.添加docker源
sudo add-apt-repository \
"deb [arch=amd64] https://mirrors.ustc.edu.cn/docker-ce/linux/ubuntu \
$(lsb_release -cs) \
stable"
5.安装docker-ce
sudo apt-get update
sudo apt-get install docker-ce
6.启动docker
sudo systemctl enable docker
sudo systemctl start docker
7.测试docker是否安装成功
sudo docker run hello-world
三、nvidia-docker 安装
1.添加源 https://blog.csdn.net/weixin_39610043/article/details/85334877
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \ sudo apt-key add - curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu16.04/amd64/nvidia-docker.list | \ sudo tee /etc/apt/sources.list.d/nvidia-docker.list sudo apt-get update
2.安装nvidia-docker2软件包
sudo apt-get install nvidia-docker2
如果提示类似下面错误:
The following packages have unmet dependencies:
nvidia-docker2 : Depends: docker-ce (= 17.06.2~ce-0~ubuntu) but 18.06.3~ce~3-0~ubuntu is to be installed or
docker-ee (= 17.06.2~ee-0~ubuntu) but it is not installable
E: Unable to correct problems, you have held broken packages.
查看docker支持版本
apt-cache madison docker-ce
安装docker低版本
sudo apt-get install docker-ce=17.06.2~ce-0~ubuntu
再重新安装指定版本nvidia-docker2
apt-get install nvidia-docker2=2.0.3+docker17.06.2-1 nvidia-container-runtime=2.0.0+docker17.06.2-1 -y
3.重新加载Docker守护程序配置
sudo pkill -SIGHUP dockerd
4.测试
sudo docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi