1.查看显卡驱动
在终端中输入如下命令:
nvidia-smi
可以看到 Driver Version(显卡的驱动版本) 和支持的 CUDA Version(所支持的CUDA版本上限)
2.安装对应版本的CUDA
根据1中的CUDA Version选择合适版本的CUDA
https://developer.nvidia.com/cuda-toolkit-archive
然后选择目标平台(我这里以X86架构上的ubuntu22.04为例,使用runfile(local)方式进行下载)
最后,输入Base Installer中提供的Installation Instructions即可
3.安装界面
(1)选择continue
(2)按回车取消 Driver 的安装,然后选择install
(3)等待安装完成,完成后注意有一行
ToolKit: Installed in /usr/local/cuda-*/
后面修改环境变量时要使用
4.添加CUDA的环境变量
sudo vim ~/.bashrc
在其中添加(cuda-12.2换为对应版本,在安装界面):
export PATH=/usr/local/cuda-12.2/bin:$PATH
使用source命令使其生效
source ~/.bashrc
可以使用 nvcc --version 命令查看版本信息来检查安装是否成功