在 CentOS 7/8 上安装 NVIDIA Container Toolkit


参考官方文档,在 CentOS 7 上安装: https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html#id1

设置存储库和 GPG 密钥

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

更新包列表后安装 nvidia-container-toolkit 包(和依赖项)

sudo yum clean expire-cache
sudo yum install -y nvidia-container-toolkit

配置 Docker 守护进程以识别 NVIDIA 容器运行时

sudo nvidia-ctk runtime configure --runtime=docker

设置默认运行时后,重新启动 Docker 守护进程以完成安装

sudo systemctl restart docker

测试

此时,可以通过运行基本 CUDA 容器来测试工作设置:

sudo docker run --rm --runtime=nvidia --gpus all nvidia/cuda:11.6.2-base-ubuntu20.04 nvidia-smi

也可以加入 -L 参数,只显示显卡:

sudo docker run --rm --runtime=nvidia --gpus all nvidia/cuda:11.6.2-base-ubuntu20.04 nvidia-smi -L

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/qq_35395195/article/details/131413664