我们使用Docker的过程中往往需要使用GPU,于是需要安装NVIDIA Driver、NVIDIA Container Toolkit和CUDA Toolkit等工具。
如果你使用的是Ubuntu
等Linux原生系统 + docker
:你需要在 /etc/docker/daemon.json
中设置default-runtime
为nvidia
,然后重启docker
,确保你启动了NVIDIA Container Toolkit。文章来源:https://www.toymoban.com/news/detail-722226.html
{
"runtimes": {
"nvidia": {
"path": "nvidia-container-runtime",
"runtimeArgs": []
}
},
"default-runtime": "nvidia"
}
但是如果你使用的是WSL 2.0
+ Docker Desktop
,你会发现根本没有 /etc/docker/daemon.json
这个路径。那么这时你需要在Docker Desktop
中点击右上角的Setting
,然后点击Docker Engine
,在这里设置完上述配置后,点击restart
,以确保你启动了NVIDIA Container Toolkit。
By the way ~ WSL 2
也需要在Resources
中设置确保开启。文章来源地址https://www.toymoban.com/news/detail-722226.html
到了这里,关于docker: Error response from daemon: unknown or invalid runtime name: nvidia 解决方案的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!