在完成 Nvidia 驱动的安装之后,接下来进行 CUDA 的安装以及版本确认。
一、安装 CUDA
1、运行 CUDA 安装程序
/mnt/data/Nvidia/CUDA# ./cuda_12.1.0_530.30.02_linux.run
在安装过程中,确保不要选择安装驱动,以免覆盖已经安装好的 Nvidia 驱动版本。
二、确认 CUDA 环境变量
1、检查环境变量是否设置成功
确认环境变量是否存在,否则需要将环境变量添加才可以使用相关命令。
root@New-test2:/mnt# cat /etc/profile
确认文件中是否包含了上述设置的 PATH 和 LD_LIBRARY_PATH 环境变量。
如果不包含,通过编辑 /etc/profile 文件来添加以下内容:
export PATH=/usr/local/cuda-12.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
2、使用 nvcc --version 命令来检查 CUDA 的版本,确保安装正确
nvcc -V