背景:原先用的CUDA11.4,但是mamba需要较高版本的CUDA,所以准备升级为CUDA12.0
step 1: 查看支持的最高CUDA版本
nvidia-smi
step 2: 去官网查找所需版本的CUDA的安装命令:
https://developer.nvidia.com/cuda-toolkit-archivehttps://developer.nvidia.com/cuda-toolkit-archive
先选择自己需要的版本,按照系统情况进行选择,分别在终端运行两条终端命令。
step 3: 运行安装命令
输入第①条命令:
输入第②条命令:
(如果不能使用‘sudo’命令,直接去掉即可,如上图第二个红框)
不勾选‘driver’,按空格即可
step 4:添加环境变量
vim ~/.bashrc
按一下“i”,输入:(cuda-xx.x替换成下载版本号;libxx替换为自己系统。可以直接复制终端中的路径,见红框)
export PATH=/usr/local/cuda-xx.x/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-xx.x/libxx:$LD_LIBRARY_PATH
按‘esc’退出编辑,‘:w’ 保存
输入:
source ~/.bashrc
检查是否安装成功:
nvcc -V