自动安装GPU驱动及CUDA（推荐）
更新时间：2025-08-21
GPU的驱动和CUDA是使用GPU计算的必备组件。您在使用GPU云服务器的过程中，可实现自动为GPU实例安装驱动及CUDA。
通过控制台勾选自定义GPU驱动安装驱动
创建GPU实例
请参考创建实例，并按照向导选择安装GPU驱动。该操作会自动帮助您安装GPU驱动、CUDA、Cudnn、DCGM以及Fabric manager（如果GPU包含NVSwitch）。
重装GPU实例
您可通过重装实例为已经运行中的实例安装或更新所需要的GPU驱动，操作如下：
登录云服务器控制台，并为需要安装或者更新GPU驱动的实例点击重装。
在弹窗中选择需要的公共镜像的操作系统，并勾选安装GPU驱动，选择需要的版本。
通过控制台或API的数据注入安装驱动
在创建、重装实例时可以通过注入如下的安装脚本实现安装GPU驱动：
API数据注入参数：
控制台数据注入入口：
脚本内容，替换如下脚本中的DRIVER_VERSION，CUDA_VERSION，CUDNN_VERSION参数为所需要的版本号，建议您参考建议安装的GPU驱动列表选择版本：
Plain Text
1#!/bin/bash
2DRIVER_VERSION="535.216.03"
3CUDA_VERSION="12.5.1"
4CUDNN_VERSION="9.6.0"
5WORK_DIR="/root/auto_install"
6SCRIPT_URL="http://mirrors.baidubce.com/nvidia-binary-driver/api/auto_install.sh"
7
8mkdir ${WORK_DIR}
9pushd ${WORK_DIR}
10for ((i=0; i<120; i++))
11do
12 wget --timeout=10 -t 10 ${SCRIPT_URL}
13 if [ $? -eq 0 ]; then
14 break
15 else
16 sleep 1
17 fi
18done
19bash ${WORK_DIR}/$(basename ${SCRIPT_URL}) ${DRIVER_VERSION} ${CUDA_VERSION} ${CUDNN_VERSION}
20popd
21rm -rf ${WORK_DIR}
22
23cmdline=$(cat /proc/cmdline)
24if [[ "${cmdline}" =~ "pci=realloc" ]]; then
25 echo "remove 'pci=realloc' cmdline arg and update grub"
26 default_grub_arg="/etc/default/grub"
27 sed -i 's/pci=realloc//g' ${default_grub_arg}
28 if command -v grub2-mkconfig; then
29 efi_grub_cfg=/boot/efi/EFI/centos/grub.cfg
30 if [ -f /boot/efi/EFI/rocky/grub.cfg ]; then
31 efi_grub_cfg=/boot/efi/EFI/rocky/grub.cfg
32 fi
33 grub2-mkconfig -o $efi_grub_cfg
34 fi
35 if command -v update-grub; then
36 update-grub
37 fi
38 reboot
39else
40 echo "there is no 'pci=realloc' arg in current cmdline, do nothing"
41fi查看自定义GPU驱动安装进展
在实例状态变为运行中后，登录实例可通过以下命令查看当前安装GPU驱动的进展：
Plain Text
1cat install_info.log看到如下的安装提示后，可通过nvidia-smi检查驱动是否正常工作。
Plain Text
1注意：安装过程中请避免执行重启实例、重装实例等涉及实例关机的操作，否则安装无法完整执行配置BCM事件通知
如您的业务系统需要自动化获取GPU驱动安装状态，可订阅BCM事件中的驱动安装成功并配置报警策略实现，系统将在实例中的GPU驱动安装完成后推送此事件。
