REHL8安装nvidia container toolkit
时间: 2025-01-03 13:39:31 浏览: 7
### 如何在 Red Hat Enterprise Linux 8 (RHEL 8) 上安装 NVIDIA Container Toolkit
#### 准备工作
确保系统已经更新到最新的状态并启用了必要的仓库。对于 RHEL 8,可以通过订阅管理工具来启用额外的软件库。
```bash
sudo subscription-manager repos --enable codeready-builder-for-rhel-8-x86_64-rpms
```
#### 安装 Docker 或其他兼容容器运行时
如果尚未安装Docker或其他OCI兼容的容器引擎,则需要先完成这一步骤。这里以Podman为例:
```bash
sudo yum install podman -y
```
#### 配置 Nvidia Repository 和 GPG Key
接下来配置Nvidia官方存储库以及导入GPG密钥以便验证下载包的真实性。
```bash
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | sudo tee /etc/yum.repos.d/nvidia-docker.repo
sudo rpm --import https://nvidia.github.io/nvidia-container-runtime/gpgkey
```
#### 安装 NVIDIA Container Toolkit 组件
现在可以从已设置好的源中安装所需的组件了。
```bash
sudo yum install nvidia-driver nvidia-container-toolkit -y
```
#### 设置环境变量
为了让 Podman 能够识别新的 GPU 支持,在 `/etc/containers/registries.conf` 文件里添加如下内容:
```ini
unqualified-search-registries = ["docker.io"]
...
[engine]
env = [
"NV_GPU=all",
]
```
#### 测试安装是否成功
最后通过拉取一个带有 CUDA 库的基础镜像来进行测试,确认一切正常运作。
```bash
podman run --rm --gpus all nvcr.io/nvidia/k8s/cuda-sample:nbody nbody -gpu -benchmark
```
阅读全文