马上注册,结交更多好友,享用更多功能,让你轻松玩转社区。
您需要 登录 才可以下载或查看,没有账号?立即注册
×
在 Ubuntu 20.04 上装 CUDA 12.1,流程其实很固定:先确认 GPU 驱动在不在,再拉官方本地安装包,跑安装器,最后把 PATH 和 LD_LIBRARY_PATH 写进环境变量。这篇把每一步的命令和两种用户home的区别都列清楚,照着敲基本一次过。
一、先确认 GPU 驱动
安装前第一步是看机器有没有 NVIDIA 驱动、驱动版本够不够,用这个命令:
nvidia-smi
能正常打印出显卡型号、显存和驱动版本表格,说明驱动没问题,可以往下走。如果提示命令不存在,得先装或升级 NVIDIA 驱动;CUDA 12.1 对应的驱动版本号在 530.30 及以上,装之前可以先核一下。
二、下载 CUDA 12.1 本地安装包
用 wget 拉官方的本地安装器(runfile 形式,一条命令装全套):
wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run
下载完成后建议先看一眼文件大小和 .sha256 校验值是否对得上,避免拉到不完整的文件导致安装器中途断掉。
三、执行安装
bash cuda_12.1.0_530.30.02_linux.run
跑起来之后会进一个文本界面:先同意协议,然后注意弹出来的驱动提示——如果你的驱动已经是新版,选「继续」跳过驱动安装,只装 CUDA Toolkit;如果机器本来没驱动,可以在这里一并装。装完默认路径是 /usr/local/cuda-12.1。安装耗时会看显卡和磁盘,几分钟到十几分钟都正常。
四、配置环境变量(root 和普通用户都要)
装完不配环境变量,命令行里还是找不到 nvcc。两个位置要分开写。
4.1 root 用户
vim ~/.bashrc
export PATH=/usr/local/cuda-12.1/bin PATH
export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64 LD_LIBRARY_PATH
4.2 普通用户
vim /home/xxx/.bashrc
export PATH=/usr/local/cuda-12.1/bin PATH
export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64 LD_LIBRARY_PATH
把 xxx 换成你自己的用户名。写入之后记得让配置生效:
source ~/.bashrc
4.3 验证装好了没
nvcc -V
nvidia-smi
nvcc -V 能打印出 release 12.1 的版本信息,就说明工具链通了。
五、常见问题
nvcc: command not found:环境变量没 source,或者写进了别的 shell 配置文件(比如用了 zsh 就写 ~/.zshrc)。
- 运行程序报找不到
libcuda.so:检查 /usr/local/cuda-12.1/lib64 是否在 LD_LIBRARY_PATH 里,还要注意 /etc/ld.so.conf.d 里是否注册了该路径。
- 安装器报驱动冲突:先
sudo nvidia-uninstall 或 apt remove nvidia-* 清理旧驱动,再重跑。
- 磁盘空间不够:CUDA Toolkit 装完通常占 3–5 GB,拉包之前先 df 看一下。
整套流程走下来就是「查驱动 → 拉 runfile → bash 安装 → 写环境变量 → source 验证」五步,Ubuntu 20.04 和 18.04 基本通用,只有下载链接里的版本号要跟着你想装的 CUDA 版本换。
qidao123.com企服评测-应用市场 编辑提示 |