驱动重装
在ubuntu虚拟机上进行以下操作:
- 卸载现有驱动:
sudo apt --purge remove nvidia* sudo apt autoremove - 通过
ubuntu-drivers autoinstall自动安装或从官网下载对应版本驱动 - 安装NVIDIA容器工具包,以实现Docker对GPU的调用
# 添加仓库 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 # 安装工具包,以实现Docker对GPU的调用 sudo apt-get update sudo apt-get install -y nvidia-container-toolkit sudo systemctl restart docker - 编辑/etc/docker/daemon.json文件
{ "default-runtime": "nvidia", "runtimes": { "nvidia": { "path": "/usr/bin/nvidia-container-runtime", "runtimeArgs": [] } } } - 执行sudo systemctl restart docker重启服务
- 安装完成后验证:nvidia-smi
- 最后重启系统使更改生效

