这个错误
- could not select device driver "" with capabilities: [[gpu]]
复制代码 通常是由于 Docker 没有正确识别到 GPU,大概 NVIDIA Docker 设置不正确。
一、检查主机 GPU 和 NVIDIA 驱动是否正常工作
执行
确认主机的 GPU 运行环境是否正常
如果下令输出错误,必要先确保 NVIDIA 驱动正确安装
二、检查 NVIDIA 容器工具包是否安装
执行
- dpkg -l | grep nvidia-container-toolkit
复制代码 确认 NVIDIA 容器工具包是否已正确安装
如果没有任何信息出现,则利用以下下令安装
- 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
- sudo apt update
- sudo apt install -y nvidia-container-toolkit
- sudo systemctl restart docker
复制代码 三、Docker运行时利用 NVIDIA设置
- sudo vim /etc/docker/daemon.json
复制代码 添加以下内容(如果已有其他设置,保存其他设置):
- {
- "default-runtime": "nvidia",
- "runtimes": {
- "nvidia": {
- "path": "nvidia-container-runtime",
- "runtimeArgs": []
- }
- }
- }
复制代码 编辑完长这样
在vim界面,输入:wq退出编辑状态
四、生存后重启 Docker
- sudo systemctl restart docker
复制代码 删除之前的容器,重新启动容器,及可调用GPU
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。 |