马上注册,结交更多好友,享用更多功能,让你轻松玩转社区。
您需要 登录 才可以下载或查看,没有账号?立即注册
×
一、先判断你的机器能不能装
装 CUDA 之前先确认三件事:显卡是不是 NVIDIA 卡、驱动与 CUDA 版本是否匹配、系统是不是支持(Ubuntu 20.04 是最常见的组合)。
lspci | grep -i nvidia
nvidia-smi
uname -m
cat /etc/os-release
nvidia-smi 能出表就说明驱动已在;看不出 CUDA 版本列(右上角 CUDA Version)代表驱动太旧,要先升级驱动。
nvidia-smi --query-gpu=name,driver_version --format=csv
二、安装方式怎么选
- .run 安装包:官方
cuda_11.8.0_520.61.05_linux.run,一次装驱动加工具库,最省事,但会覆盖已有驱动。
- apt 装:先设 CUDA 源再
apt install cuda-toolkit-11-8,升级管理方便,推荐生产用。
- 容器:宿主只装驱动,运行时用 NVIDIA 容器镜像,最干净。
三、apt 方式完整步骤
添加 NVIDIA 源:
sudo wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt update
装 toolkit(会连带装推荐驱动):
sudo apt install -y cuda-toolkit-11-8
只装驱动(已有驱动只想补 CUDA):
sudo apt install -y cuda-driver-11-8
配置环境变量,写进 ~/.bashrc:
export PATH=/usr/local/cuda-11.8/bin PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64 LD_LIBRARY_PATH
source ~/.bashrc
nvcc --version
nvcc 能输出 11.8 就算装好了。
四、run 包方式
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
sudo sh cuda_11.8.0_520.61.05_linux.run
交互界面里如果之前装过驱动,记得取消勾选「Driver」,只装工具包,免得驱动被反复覆盖。装完同样要配 PATH 和 LD_LIBRARY_PATH。
五、验证真的能算
nvidia-smi
nvcc -V
跑官方样例:
cd /usr/local/cuda-11.8/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
最后一行出现 Result = PASSED 就完全没问题。
六、常见故障
nvidia-smi` command not found:驱动没装或装失败,apt install -y nvidia-driver-535 后再看。
- 内核模块加载失败:刚升级过内核,secure boot 开着会拒绝加载第三方模块,进 BIOS 关安全启动,或装完重启再试。
- 版本不匹配:驱动只支持到 CUDA 11.4,你装 11.8 会报 unsupported,去看
nvidia-smi 表格里标注的最高 CUDA 版本再定。
- 编译报找不到 cuda.h:没装
cuda-toolkit 只有驱动,或 PATH 没刷新。
- 容器里看不到卡:宿主装了驱动但没装
nvidia-container-toolkit,容器要加 --gpus all。
- 显存被占满:之前的进程没退出,
fuser -v /dev/nvidia* 查是谁在占,再 kill。
qidao123.com企服评测-应用市场 编辑提示
本文按 Ubuntu 20.04 环境整理 CUDA 11.8 的两种安装方式(apt 源与 run 包)、驱动版本匹配原则、环境变量配置与验证方法,并列出六类高频故障。安装前请先确认显卡型号与 nvidia-smi 支持的最高 CUDA 版本;生产环境建议容器化隔离,宿主只保留驱动。企业 GPU 训练机可把驱动与 CUDA 版本固化进镜像基线,避免每次重建环境踩同样的坑。
|