云服务器装 NVIDIA 显卡驱动和 CUDA,这事说难不难,但坑确实不少。尤其现在很多云厂商推出了带 GPU 的实例(比如阿里云的 ecs.gn6i 系列、腾讯云的 GN 系列),不少朋友拿到手第一件事就是装驱动、跑深度学习框架,结果发现要么驱动版本和内核不匹配,要么 CUDA 装完 nvidia-smi 正常但 PyTorch 就是报错。这篇我就把从裸机到 CUDA 可用状态的完整过程拆开讲,包括每一步背后的原理、选型逻辑和排错经验,照着做基本能一次走通。
1. 环境确认与安装方案选型
1.1 先搞清楚你的 GPU 实例到底是什么架构
很多人在云服务器上装驱动翻车,第一个原因就是没搞清楚自己的实例到底是不是“真 GPU”机器。云服务器厂商对 GPU 实例的虚拟化方式不太一样,常见的分三类:
- GPU 直通(Passthrough):整个物理 GPU 直接映射给虚拟机,
lspci | grep -i nvidia能看到完整的显卡信息,这种最省心,跟物理机一样装驱动就行。 - GPU 虚拟化(vGPU):一颗物理 GPU 切成多个虚拟 GPU(比如 A10 切出来的 vGPU),这种必须用云厂商提供的驱动包,自己跑去 NVIDIA 官网下载通用驱动大概率装不上,或者装上后
nvidia-smi报No devices were found。 - MIG 模式:A100/H100 这类卡上的 MIG 切片,通常也是需要通过厂商的镜像或驱动包来管理。
所以我建议你在动手之前,先登录云厂商的控制台,看一下实例规格详情里有没有标注“GPU 直通”或“虚拟化类型”。如果是抢占式实例或共享型 GPU 实例,最好提个工单确认一下驱动安装方式。另外,不同云厂商都会维护自己的 GPU 驱动镜像,比如阿里云的“GPU 加速型”公共镜像、腾讯云的“GPU 云服务器”专用镜像,这些镜像一般已经预装好匹配的驱动和 CUDA,直接创建实例是最省事的。但既然你是要手动装,那就要先确认环境。
1.2 拿到机器后的四件事:查型号、查系统、查内核、查 gcc
不管你是新开的实例还是别人交接的机器,先执行下面这几条命令,把家底盘清楚:
bash复制# 查看 GPU 硬件信息
lspci | grep -i nvidia
# 查看发行版和版本号
cat /etc/os-release
# 查看内核版本(Ubuntu/Debian)
uname -r
# 查看 gcc 版本
gcc --version
这几条命令的结果直接决定你后面选哪个驱动文件。举个例子,Ubuntu 22.04 默认内核是 5.15,但如果你开了 HWE 内核,版本可能变成 6.2 甚至 6.5。NVIDIA 驱动是以后端内核模块(nvidia.ko)方式工作的,内核版本一变,驱动模块就要重新编译,所以驱动安装包最好选 .run 格式,方便后续对着内核版本重新安装。我已经踩过太多次“用 apt 装的驱动在系统更新内核后直接失效”的坑,后面会详细说。
另外 gcc 版本也很关键。CUDA 工具链里的 nvcc 和部分编译脚本依赖 gcc,如果你系统里连 gcc 都没装,后面编译内核模块会直接报 gcc: command not found。Ubuntu 上装依赖:
bash复制sudo apt update
sudo apt install -y build-essential dkms linux-headers-$(uname -r)
这里解释一下这几个软件包的作用:
build-essential:包含 gcc、g++、make 等编译工具链,编译驱动模块必需。dkms:Dynamic Kernel Module Support,它能让驱动模块在系统更新内核后自动重新编译,这个一定要装。linux-headers-$(uname -r):当前内核的头文件,编译内核模块时必须有对应的头文件,不然会报找不到linux/version.h之类的错误。
注意:如果
apt update报错,先看一下是不是源的问题。国内云服务器建议把 apt 源改成阿里云或清华源,不然下载速度让人崩溃。改源的方法是把/etc/apt/sources.list里的地址替换成mirrors.aliyun.com或mirrors.tuna.tsinghua.edu.cn,具体按发行版格式来。
1.3 驱动安装的三种方式:推荐哪个,为什么
NVIDIA 驱动在 Linux 下的安装方式主流有三种,我列个表格对比下适用场景:
| 方式 | 优点 | 缺点 | 适用场景 |
|---|---|---|---|
发行版仓库安装(apt install nvidia-driver-XXX) |
与系统集成好,自动处理依赖 | 版本较旧,内核升级后偶尔出问题 | 不跑最新框架、求稳的人 |
NVIDIA 官网 .run 文件安装 |
版本最新,可自定义安装项 | 需要自己管理后续升级和依赖 | 需要特定版本驱动、跑新卡 |
| 云厂商预置驱动镜像 | 与云平台虚拟化兼容性最好 | 灵活性差,版本受厂商控制 | 商业环境、不想折腾 |
我的建议是:如果是 V100、T4、A10 这些云上常见的卡,首选 .run 文件安装。为什么?因为云服务器的内核是云厂商定制的,用 apt 源里的驱动版本可能匹配不上,而且 apt 版本普遍偏旧。比如你要装 CUDA 12.4,官方要求驱动版本至少要 550.54.14,Ubuntu 22.04 自带的软件源里很可能只有 535 或 470,这样就带不动新 CUDA。
如果你确定自己对版本不敏感,只是想把 nvidia-smi 跑起来,那用 ubuntu-drivers devices 自动推荐也不丢人:
bash复制sudo ubuntu-drivers devices
sudo apt install -y nvidia-driver-535
但你既然是为了跑 CUDA 相关的内容,我还是建议走官网 .run 流程。因为当你需要在同一台机器上切换多个 CUDA 版本时,.run 加环境变量的方式最灵活。下面整个流程就以 .run 为主线来写。
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. NVIDIA 驱动安装实操与卸载避坑
2.1 下载匹配型号的驱动:去官网别乱搜
打开 NVIDIA 驱动下载页面(NVIDIA Driver Download),按你的 GPU 型号、操作系统、语言、版本筛选。以 Ubuntu 22.04 + T4 为例,一般选:
- Product Type: Data Center / Tesla
- Product Series: T-Series
- Product: Tesla T4
- Operating System: Linux 64-bit
- CUDA Toolkit: 选你需要的版本,或者选 All
下载完是个 .run 文件,比如 NVIDIA-Linux-x86_64-550.54.14.run。这里有个细节:如果你在国内服务器上直接下载,NVIDIA 官网速度可能很慢。可以先去自己的电脑浏览器下载好,再 scp 传到服务器;或者用云厂商提供的内网镜像源。我之前在阿里云上海区下载 NVIDIA 驱动,用 wget 从官网拉只有几十 KB/s,后来改用本机下载再传上去,几分钟搞定。
2.2 卸载旧驱动:不卸载干净等于白干
如果你之前装过驱动(无论 apt 还是 run 方式),建议先清理干净,否则新驱动装上后可能有冲突。完整的卸载命令:
bash复制# 用 run 文件卸载(如果有)
sudo nvidia-uninstall
# 清理 apt 安装的驱动
sudo apt-get purge -y nvidia-*
sudo apt-get autoremove -y
然后务必检查一下还有没有残留的 nvidia 模块:
bash复制lsmod | grep nvidia
如果有输出,说明模块还在占用,需要先停掉相关服务或者 sudo modprobe -r nvidia_uvm nvidia_drm nvidia_modeset nvidia 逐个移除,顺序是从依赖到被依赖。不建议硬删,容易把系统搞崩。
另外,很多人不知道的是:NVIDIA 驱动装上后会在内核里注册 DRM 模块(Direct Rendering Manager),如果你之前开过 Wayland 显示服务,在卸载驱动的时候可能会遇到图形界面崩溃的情况。但云服务器基本都是纯命令行,所以这个坑在云上不常见,物理机上要小心。
2.3 run 文件安装全流程:从禁用 nouveau 到 reboot
NVIDIA 驱动和系统开源的 nouveau 驱动是不共戴天的,两者都在抢占同一个 GPU 设备节点。所以安装英伟达闭源驱动的第一步,是把 nouveau 禁用掉。操作流程:
先创建黑名单文件:
bash复制sudo vim /etc/modprobe.d/blacklist-nouveau.conf
写入以下内容:
bash复制blacklist nouveau
options nouveau modeset=0
然后更新 initramfs:
bash复制sudo update-initramfs -u
此时 必须重启,让 nouveau 不再加载。重启后确认:
bash复制lsmod | grep nouveau
没有任何输出就说明禁用成功。如果还有输出,说明黑名单没生效,检查一下是不是有别的文件也加载了 nouveau(比如某些云厂商的定制内核会在 /etc/modprobe.d/ 下面放自己的配置,覆盖了你的设置)。
确认干净之后,给 run 文件加执行权限并运行:
bash复制chmod +x NVIDIA-Linux-x86_64-550.54.14.run
sudo ./NVIDIA-Linux-x86_64-550.54.14.run
安装界面注意几个选项:
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 550.54.14?选 YesInstall NVIDIA's 32-bit compatibility libraries?按需选。跑深度学习可以直接 No,省点空间。Would you like to run the nvidia-xconfig utility?选 No。云服务器没有显示器,写 xorg 配置反而容易出错。- 如果提示需要编译内核模块,会调用 dkms,确认 gcc 和内核头文件已经装好就行。
安装完成后,再次重启(有些场景不重启也能用,但为了保险建议重启),然后验证:
bash复制nvidia-smi
如果看到类似下面的输出,说明驱动已经正常工作:
text复制+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 550.54.14 Driver Version: 550.54.14 CUDA Version: 12.4 |
+---------------------------------------------------------------------------------------+
这里的 CUDA Version 是驱动所支持的最高 CUDA 版本,不代表你已经装好了 CUDA Toolkit。这个很多人会误会,后面细说。
2.4 驱动验证失败:核心问题排查思路
nvidia-smi 报 command not found,通常有三种原因:
- 驱动装失败了。
- 驱动装好了,但
/usr/bin/nvidia-smi不在 PATH 里(run 文件默认装到/usr/bin,但有时会放/usr/local/bin)。 - 模块没有加载。
排查顺序:
bash复制# 1. 看模块加载情况
lsmod | grep nvidia
# 2. 看设备节点
ls /dev/nvidia*
# 3. 找 nvidia-smi 位置
find / -name nvidia-smi 2>/dev/null
如果 lsmod 没有 nvidia 模块,多半是内核模块编译失败。去 /var/log/nvidia-installer.log 里翻错误记录。最常见的错误是 unable to load the 'nvidia-drm' kernel module,这种时候十有八九是内核头文件版本和当前内核不一致。uname -r 返回什么版本,/usr/src/linux-headers-$(uname -r) 目录必须存在,缺了就重新装一遍 linux-headers-$(uname -r)。
另外,有些云服务器开启了 Secure Boot,这会导致第三方内核模块无法加载。确认方法:
bash复制mokutil --sb-state
如果输出 SecureBoot enabled,那你就得去 BIOS 里关掉,或者在安装驱动的最后一步选择 Sign kernel module 并注册密钥。但绝大多数云服务器默认是关闭 Secure Boot 的,这里只是提醒物理机或者某些虚拟化平台会遇到。
3. CUDA Toolkit 安装与多版本共存管理
3.1 先理清楚驱动和 CUDA 的关系
很多人一开始搞不明白 nvidia-smi 显示的 CUDA Version 和 nvcc --version 显示的 CUDA Version 为什么不一样。简单说:
- 驱动里内置的 CUDA Driver API:决定了你的显卡最高能用哪个 CUDA 版本。比如驱动 550.54.14 显示 CUDA Version 12.4,意思是它能支持 CUDA 12.4 及更低版本的 runtime。
- CUDA Toolkit 里的
nvcc编译器:真正负责把代码编译成 GPU 指令的工具。你跑 PyTorch 时,PyTorch 自带的 CUDA runtime 只要不高于驱动支持的上限,就能正常跑。
所以,nvidia-smi 的 CUDA Version 只是驱动对外的能力声明,并不能替代 CUDA Toolkit 的安装。有人说“我 nvidia-smi 已有 CUDA 12.4,是不是不用装了?”这是不对的。你要用 nvcc 编译 CUDA 代码,或者某个框架依赖特定 CUDA 工具链,就必须装 Toolkit。
3.2 CUDA Toolkit 下载与安装:repo 还是 runfile?
去 NVIDIA CUDA Toolkit 下载页面,选择操作系统和架构,NVIDIA 会给你两种安装方式:
deb (network)或deb (local)方式:通过 apt 源安装,好处是后续可以用 apt 管理依赖和升级。runfile (local)方式:一个巨大的.run文件,包含全套 Toolkit 和驱动,但你可以选择只装 Toolkit 部分。
我的建议是:如果你已经在前面手动装好了驱动,CUDA Toolkit 一定不要选包含驱动的方式,用 runfile 里的自定义选项,只勾选 CUDA Toolkit 本身,把 Driver 选项关掉。不然它会把你的驱动降级或覆盖成 CUDA 包里自带的版本,搞不好版本比你现在装的还要旧,然后整个环境就混乱了。
以 CUDA 12.4 为例,runfile 方式是:
bash复制wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run
sudo sh cuda_12.4.0_550.54.14_linux.run
注意文件名里 _550.54.14 其实指的是这个安装包内置的驱动版本。进入安装界面后,用方向键上下移动,把 Driver 前面的 [X] 改为 [ ],然后选 Install。这样装的就只有 CUDA Toolkit,包括 nvcc、cudart、cublas、cudnn 的 dev 包等。
3.3 环境变量配置:让 nvcc 乖乖听话
安装完成后,默认 CUDA 会装到 /usr/local/cuda-12.4,并且系统会创建一个软链接 /usr/local/cuda 指向它。你需要把 bin 和 lib64 加进 PATH 和 LD_LIBRARY_PATH。在 ~/.bashrc 末尾加:
bash复制export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
然后:
bash复制source ~/.bashrc
nvcc --version
输出显示 Cuda compilation tools, release 12.4, V12.4.131 之类的信息就成功了。
这里有个细节:不建议直接把 /usr/local/cuda 这个软链接路径写死,而是应该用 /usr/local/cuda-12.4 写进环境变量。因为后面如果你装了多个 CUDA 版本,你只需要改软链接指向,而不必改环境变量(环境变量里用软链接路径即可)。这是多版本管理的基础。
3.4 多版本 CUDA 共存:glibc 兼容性和软链接切换
实际开发中,多版本共存的诉求非常常见。比如你既要跑 PyTorch 2.1(依赖 CUDA 12.1),又要跑某个自定义的 CUDA C++ 项目(需要 CUDA 11.8),这时候最忌讳的就是装完一个新版本把旧版本卸了、重新配环境变量。
正确做法是:
- 多个 CUDA 版本安装在不同目录:默认情况下,runfile 会根据版本号装到
/usr/local/cuda-11.8、/usr/local/cuda-12.1、/usr/local/cuda-12.4,互不干扰。 - 用软链接切换默认版本:
bash复制sudo rm -rf /usr/local/cuda sudo ln -s /usr/local/cuda-12.1 /usr/local/cuda - 如果某个项目需要指定别的版本,在项目构建脚本里临时覆盖环境变量即可:
bash复制export PATH=/usr/local/cuda-11.8/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH
但注意,下山的 GLIBC 版本是全局的。如果你系统是 Ubuntu 22.04,glibc 版本是 2.35,老一点的 CUDA 11.4 及以下版本可能对 glibc 版本有额外要求。如果遇到报错 GLIBC_2.34 not found 这种,实际上不是 CUDA 的问题,而是编译某个动态库的环境问题。这时建议就别往下折腾了,直接升级 CUDA 版本,或者用容器镜像(NVIDIA NGC 容器)来跑老版本环境。
另外关于 update-alternatives:Ubuntu 上也可以用 update-alternatives 来管理 CUDA 软链接,但对 CUDA 这种简单路径的需求来说,手动维护软链接更直观,不容易把系统搞乱。
3.5 cuDNN 安装:深度学习跑起来的关键一步
如果你要跑深度学习框架,cuDNN 几乎是必装的。云服务器上装 cuDNN 有两种办法:
- 从 NVIDIA 官网下载 deb 包。需要注册 NVIDIA Developer 账号,下载
cudnn-linux-x86_64-8.9.x.7_cuda12-archive.tar.xz(对应 CUDA 12.x)。解压后把里面的 include 和 lib 复制到 CUDA 目录:bash复制tar -xvf cudnn-linux-x86_64-8.9.7.29_cuda12-archive.tar.xz sudo cp cudnn-linux-x86_64-8.9.7.29_cuda12-archive/include/cudnn*.h /usr/local/cuda/include/ sudo cp cudnn-linux-x86_64-8.9.7.29_cuda12-archive/lib/libcudnn* /usr/local/cuda/lib64/ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn* - 用 pip 安装。如果你只是跑 PyTorch/TensorFlow,可以直接在 Python 环境里装
nvidia-cudnn-cu12这个 pip 包,它会自动把 cuDNN 的 so 文件放进 site-packages,框架能直接加载。但这个方案不能让系统全局访问 cuDNN,适合纯 Python 项目,不适用于需要编译 CUDA C++ 代码的场景。
提示:cuDNN 版本必须和 CUDA 大版本匹配。比如 CUDA 12.1 就用 cuDNN 8.9.x for CUDA 12.x,CUDA 11.8 就用 cuDNN 8.9.x for CUDA 11.x。版本不匹配时,运行时会报各种奇怪的符号找不到错误(例如
undefined symbol: cudnnCreate),别问我怎么知道的。
4. 深度学习环境验证:驱动、CUDA、PyTorch 三端对齐
4.1 框架安装不是装个 pip 包那么简单
很多人装了驱动、装好 CUDA Toolkit,然后 pip install torch,发现 torch.cuda.is_available() 返回 False,心态直接崩了。其实这里面的核心问题是:PyTorch 的 CUDA 绑定必须和你的 CUDA 版本匹配。
PyTorch 官方提供的预编译包通常有 cu118、cu121、cu124 等版本。如果你默认 pip install torch,装到的可能是 CPU 版本(比如在某些平台上 pip 默认解析到 CPU wheel),或者是对应最低 CUDA 版本的 wheel。因此,最稳妥的安装方式是从 PyTorch 官网的安装命令区拿对应的 URL 安装,比如:
bash复制pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
这样安装的 wheel 是针对 CUDA 12.1 预编译的,和系统里的 CUDA Toolkit 版本不一定需要完全一致,但一定不能高于驱动支持的 CUDA 版本。
4.2 验证命令全家桶:从底层到应用层层把关
为了确认整个链路是通的,我通常按这个顺序验证:
bash复制# 1. 验证驱动
nvidia-smi
# 2. 验证 CUDA Toolkit
nvcc --version
# 3. 验证 CUDA 驱动 API 和 runtime 都能访问(编译一个小示例)
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
deviceQuery 结尾如果显示 Result = PASS,说明硬件、驱动、CUDA Toolkit 三者的配合没问题。这里再提醒一下,deviceQuery 这种老代码需要 make 和 gcc 版本兼容,如果编译报错,试试把 Makefile 里的 HOST_COMPILER = g++ 改成 HOST_COMPILER = g++-11(取决于你系统里有的 gcc 版本)。
接着验证 PyTorch:
python复制import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())
print(torch.cuda.get_device_name(0))
如果输出 True 和你的显卡型号,说明整条链路通了。如果 torch.cuda.is_available() 是 False,先确认你装的是不是 CUDA 版 torch:
bash复制python -c "import torch; print(torch.__config__.show())"
看看 CUDA_GENCODE 和 CUDA_HOME 是否正常。如果这时提示 libcudart.so.12: cannot open shared object file,说明系统找不到 CUDA 的 so 文件,回头检查 LD_LIBRARY_PATH。
4.3 容器方案:NVIDIA Container Toolkit 让环境折腾终结
我个人的经验是,如果只是跑深度学习框架,不要把宿主机环境当成试验田。宿主机装好驱动和 CUDA Driver 后,直接把应用跑在 Docker 容器里,工具链全部用容器镜像自带的 CUDA 运行时。这样既能实现多版本隔离,又能避免反复污染系统环境。
在云服务器上装 NVIDIA Container Toolkit 的步骤(基于 Ubuntu 22.04):
bash复制curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
sudo apt-get update
sudo apt-get install -y nvidia-container-toolkit
sudo nvidia-ctk runtime configure --runtime=docker
sudo systemctl restart docker
配置完成后,用 docker run 跑一个带 GPU 的容器验证:
bash复制docker run --rm --gpus all nvidia/cuda:12.4.0-base-ubuntu22.04 nvidia-smi
如果在容器里能正常显示 nvidia-smi 的输出,说明宿主机驱动 + 容器 GPU 透传这条链路也是通的。接下来想用哪个 PyTorch 版本,直接拉对应镜像即可,宿主机环境基本不用动了。
提示:
nvidia-ctk runtime configure针对的是 Docker。如果用 Containerd(比如 Kubernetes 场景),步骤类似,但是要指定--runtime=containerd。这个我在部署 K8s GPU 节点时踩过坑,默认只配置 Docker 的话,K8s 里的 Pod 无法申请 GPU 资源。
5. 常见问题与排查技巧实录
5.1 编译内核模块时报错
典型报错信息:
text复制Error: unable to load the 'nvidia-drm' kernel module
排查思路:
dkms status查看模块状态,看看是不是installed还是built。- 如果是
built而不是installed,说明 DKMS 只是编译了但没装进内核,手动执行sudo dkms install -m nvidia -v 550.54.14。 - 如果编译时报
Error! Bad return status for module build on kernel,把/var/lib/dkms/nvidia/550.54.14/build/make.log末尾几行打开看,通常能看到具体是缺哪个头文件。
5.2 驱动装完、CUDA 装完,torch.cuda.is_available() 还是 False
可能原因和对应的排查顺序:
| 可能原因 | 检查方法 | 解决方案 |
|---|---|---|
| 装的是 CPU 版 torch | torch.__version__ 末尾带 +cpu |
用 PyTorch 官网的 cu 指定版本重新安装 |
| CUDA 库路径没生效 | echo $LD_LIBRARY_PATH 是否包含 /usr/local/cuda/lib64 |
重新 source 并确认 |
| 驱动与 CUDA 不兼容 | nvidia-smi 的 CUDA Version 低于你装 Toolkit 的大版本 |
升级驱动,或降级 CUDA Toolkit |
| 权限问题 | 容器内普通用户访问 GPU | 给用户加 video 组:sudo usermod -aG video $USER |
我最常遇到的其实是第一种,尤其是在国内环境,某些 pip 镜像默认把 torch 解析成了 CPU 版。解决办法是用 --index-url 强制指定 PyTorch 的官方源。
5.3 Ubuntu 内核更新后 nvidia-smi 消失
这是 Ubuntu 上特别高频的问题。某天你运行 sudo apt upgrade,顺手把内核从 5.15.0-91 升级到 5.15.0-92,重启之后 nvidia-smi 报驱动找不到。原因就是新内核没有对应的 nvidia 模块。
如果你装了 DKMS,理论上它会自动重新编译,但有时因为网络或编译依赖问题,DKMS 没跑成功。手动处理:
bash复制sudo dkms status
# 如果显示 nvidia/550.54.14: 5.15.0-92-generic (built, not installed)
sudo dkms install -m nvidia -v 550.54.14
sudo modprobe nvidia
如果 dkms 状态里显示 install 失败,先看 /var/lib/dkms/nvidia/550.54.14/build/make.log,最常见的是 gcc 版本不匹配。Ubuntu 22.04 升级后有可能默认 gcc 是 11,但旧的驱动源码不支持新 gcc,需要设置 CC 环境变量指定编译用的编译器:
bash复制sudo CC=gcc-10 dkms install -m nvidia -v 550.54.14
5.4 云服务器 GPU 实例常见的性能问题:降频和掉卡
最后提一个云上独有的问题:GPU 降频。你在裸金属机上跑的 nvidia-smi 通常能看到稳定的频率,但在云服务器上,因为是共享物理资源,可能会出现 nvidia-smi 里看到 P8 这种低性能状态,或者运行大任务时 GPU 利用率不高。这种情况通常是实例规格限制了 GPU 的最大功耗或 GPU 时间片,不是驱动的问题。
解决办法是在创建实例时选择GPU独占型实例(比如阿里云的 GPU 计算型 gn6v、gn7i),而不是 GPU 共享型的实例。共享型省钱,但你要是拿它跑模型训练,性能波动会让你怀疑人生。
另外还有个排查操作用得多:watch -n 1 nvidia-smi 可以用来持续观察 GPU 利用率、显存占用、温度、功耗,判断是不是真的在跑计算。训练脚本起来了但 GPU 利用率一直在 0%,那大概率是数据加载成了瓶颈,CPU 来不及喂数据,这和驱动/CUDA 没关系,是工程优化的问题。
5.5 一个小工具整理
我每次排查 GPU 环境都会用一条命令确认整体状态:
bash复制nvidia-smi --query-gpu=name,driver_version,memory.total,compute_cap --format=csv
这个能快速看显卡的算力版本(Compute Capability)。有些老卡算力版本太低,比如 3.0,新版 PyTorch 已经不支持了,跑起来会报 no kernel image is available for execution on the device。云上主流显卡比如 T4(7.5)、V100(7.0)、A10(8.6)、A100(8.0)都还比较新,问题不大,但如果你的云厂商给你分配的是老卡型号,留意一下这个字段。
6. 个人实践总结和建议
折腾 GPU 云服务器这套环境多了以后,我最大的体会是:别把所有组件都一股脑往系统里塞,分层管理才是长期省心的关键。
具体来说:
- 宿主机上只保证驱动和 CUDA Driver 可用,也就是
nvidia-smi能出正常输出,这个版本建议保持相对新但不要追新,稳定优先。 - CUDA Toolkit 和 cuDNN 这类开发工具链,虽然可以装在宿主机上,但尽量用软链接管理多版本,或者干脆做成容器镜像。
- 应用层的 Python 依赖、PyTorch 版本、CUDA 绑定,全部用虚拟环境或 Docker 容器管理。虚拟环境(conda/venv)+ Docker 双管齐下,基本能做到一台机器同时支撑多个项目互不干扰。
最后分享一个排查环境问题时我反复用的小技巧:把每一步验证拆开,不要跳过。先确认驱动,再确认 CUDA,最后才是框架。很多人喜欢直接跑到 torch.cuda.is_available() 这一步,出错了就来回折腾,完全不知道是底层哪一环出了问题。按 nvidia-smi → nvcc --version → deviceQuery → PyTorch 的顺序来,每一层都是下一层的坚实基础,哪一层出问题,解决哪一层就好。这套方法论解决的不只是云服务器上的问题,在 WSL、物理机、Docker 容器里同样适用。
