由于使用到的库需要CUDA2.1+,本人所用的nvidia驱动版本为535,最高支持12.2,并且适配良好
所以卸载CUDA11.7并安装CUDA12.2,具体步骤如下:
一.删除CUDA11.7
# 卸载 CUDA 工具包及相关组件
sudo apt-get --purge remove "cuda*" "nvidia-cuda*"
sudo apt-get --purge remove "libcudnn*" "libnccl*" # 清理 cuDNN 等依赖
# 清理残留文件
sudo apt-get autoremove
sudo apt-get autoclean
sudo rm -rf /usr/local/cuda-11.7 /usr/local/cuda # 删除安装目录和软链接
二.安装CUDA12.2
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda-repo-ubuntu2004-12-2-local_12.2.0-535.54.03-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-12-2-local_12.2.0-535.54.03-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-12-2-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda
三.验证是否安装成功
nvcc -V # 应显示 "release 12.2, V12.2.91" 或类似版本
四.配置环境变量
配置环境变量
打开 .bashrc 文件:
nano ~/.bashrc
在文件末尾添加以下内容:
export PATH=/usr/local/cuda-12.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
使配置生效:
source ~/.bashrc
验证安装
运行以下命令验证:
nvcc --version
浙公网安备 33010602011771号