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

本文介绍如何在Docker中配置NVIDIA驱动以支持GPU加速。主要内容包括:编辑daemon.json文件设置默认运行时为nvidia;确认安装了正确的nvidia-docker或nvidia-container-toolkit版本;修改docker.service文件以支持GPU运行。
3266

被折叠的 条评论
为什么被折叠?



