Docker部署英伟达GPU应用
英伟达官方解决方案
安装Nvidia Container Toolkit,配置docker运行时,使容器可以使用Nvidia Container运行时。构建容器镜像时指定nvidia运行时、gpu(可选)即可。
通过在容器中执行nvidia-smi命令可以检测容器环境中cuda是否可用。
安装Nvidia Container Toolkit,配置docker运行时,使容器可以使用Nvidia Container运行时。构建容器镜像时指定nvidia运行时、gpu(可选)即可。
通过在容器中执行nvidia-smi命令可以检测容器环境中cuda是否可用。