hubujiejinyu

导航

 

由于使用到的库需要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

具体连接如下:https://developer.nvidia.com/cuda-12-2-0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=20.04&target_type=deb_local

三.验证是否安装成功

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

posted on 2025-10-22 17:04  hubujiejinyu  阅读(76)  评论(0)    收藏  举报