直接
ubuntu-drivers devices # 查看需要安装的驱动
sudo apt-get purge nvidia-* # 清空旧的驱动
sudo apt-get autoremove
sudo apt-get update # 更新下
sudo apt grade # 最好都更新一下
sudo apt install nvidia-driver-535 nvidia-dkms-535
然后安装nvidia-docker
sudo apt-get install nvidia-container-toolkit
sudo nvidia-ctk runtime configure --runtime=docker
sudo systemctl restart docker