我们使用Docker的过程中往往需要使用GPU,于是需要安装NVIDIA Driver、NVIDIA Container Toolkit和CUDA Toolkit等工具。

如果你使用的是Ubuntu等Linux原生系统 + docker:你需要在 /etc/docker/daemon.json 中设置default-runtime为nvidia,然后重启docker,确保你启动了NVIDIA Container Toolkit。

{

"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中设置确保开启。

精彩链接

评论可见,请评论后查看内容,谢谢!!!
 您阅读本篇文章共花了: