Linux安装NVidia Docker Runtime

NVIDIA Container Runtime 是一种 GPU 感知容器运行时,与 Docker、CRI-O 和其他流行容器技术使用的 Open Containers Initiative (OCI) 规范兼容。 它简化了构建容器化 GPU 加速应用程序并将其部署到桌面、云或数据中心的过程。
借助 NVIDIA Container Runtime 支持的容器技术(如 Docker),开发人员可以将其 GPU 加速应用程序及其依赖项打包到一个包中,无论部署环境如何,都能保证在 NVIDIA GPU 上提供最佳性能。

介绍完了我们接着直接看看安装方法

ubuntu中的安装方法如下

curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | \
sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.list | \
sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list
sudo apt-get update
sudo apt-get install -y nvidia-container-runtime

centos中的安装方法如下

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.repo | \
sudo tee /etc/yum.repos.d/nvidia-container-runtime.repo
sudo yum install nvidia-container-runtime

安装完成以后可以使用下面命令进行检查

which nvidia-container-runtime
/usr/bin/nvidia-container-runtime

然后我们需要在docker的配置文件daemon.json中添加配置

  "runtimes": {
    "nvidia": {
      "runtimeArgs": [], 
      "path": "/usr/bin/nvidia-container-runtime"
    }
  }

然后重启docker是配置生效

systemctl daemon-reload
systemctl restart docker

然后我们可以拉取一个cuda镜像进行测试

docker run --rm --runtime=nvidia nvidia/cuda:9.0-base nvidia-smi

能够正常打印nvidia-smi的信息说明咱们的nvidia-container-runtime安装成功了。

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:https://sulao.cn/post/880.html

我要评论

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。