Docker部署英伟达GPU应用

英伟达官方解决方案

安装Nvidia Container Toolkit,配置docker运行时,使容器可以使用Nvidia Container运行时。构建容器镜像时指定nvidia运行时、gpu(可选)即可。

通过在容器中执行nvidia-smi命令可以检测容器环境中cuda是否可用。

CoolCats
CoolCats
理学学士

我的研究兴趣是时空数据分析、知识图谱、自然语言处理与服务端开发