1. 检查daemon.json文件是否配置nvidia驱动
vim /etc/docker/daemon.json
{
"default-runtime": "nvidia",
"runtimes": {
"nvidia": {
"path": "/usr/bin/nvidia-container-runtime",
"runtimeArgs": []
}
}
}
2. 查看nvidia-docker(19以下) 或 nvidia-docker2 (19以下)或 nvidia-container-toolkit(19及以上)
3. 查看docker.service文件
docker.service
[Service]
ExecStart=
ExecStart=/usr/bin/dockerd --add-runtime=nvidia=/usr/bin/nvidia-container-runtime
sudo systemctl daemon-reload
sudo systemctl restart docker