更新时间:2024-09-25 18:29:24

支持配置和使用英伟达GPU

注意:本章节步骤需要在GPU节点上执行,当前适配仅支持x86环境
第1步 将nvidia驱动依赖包nvidia-driver-535.154.05-20240324.4.el8.x86_64.tar上传至环境中,解压得到同名文件夹

上传驱动安装包.png

第2步 进入解压文件夹中,执行./install.sh安装驱动和基础组件

安装驱动.png

第3步 执行cd /etc/docker/进入GPU所在节点,执行vi daemon.json,将以下内容添加到daemon.json文件中,并重启docker服务
"default-runtime": "nvidia",
"runtimes": {
"nvidia": {
"args": [],
"path": "nvidia-container-runtime"
}
}

编辑配置文件.png

编辑配置文件2.png