CUDA 驱动版本替换
- 查看显卡型号
lspci | grep NVIDIA
2.选择目标驱动
官方驱动 | NVIDIA 选择需要的驱动。
wget https://cn.download.nvidia.com/tesla/515.105.01/NVIDIA-Linux-x86_64-515.105.01.run
3.停用已有的CUDA
sudo lsmod | grep nvidia --查看哪些进程在用
sudo lsof /dev/nvidia* -- 查看这些进程用
sudo kill -9 PID -- 杀死当前进程, 如果被重复拉起,请参考https://blog.csdn.net/zhimakaimen503/article/details/108795667
杀死父进程。
sudo rmmod nvidia -- 停用nvidia
4. 安装
sudo sh NVIDIA-Linux-x86_64-515.105.01.run -no-x-check -no-nouveau-check -no-opengl-files
CUDA Toolkit 安装
- 确定版本
CUDA Toolkit 11.7 Downloads | NVIDIA Developer
11-7-0 根据cuda版本不同,可以改成不同的。
2. 下载并安装
chmod -x cuda_11.7.0_515.43.04_linux.run #增加执行权限。
sudo sh cuda_11.7.0_515.43.04_linux.run #其中driver不要选择,上步已经安装
3. 配置环境变量
#在切换cuda版本时
rm -rf /usr/local/cuda#删除之前创建的软链接
sudo ln -s /usr/local/cuda-8.0/ /usr/local/cuda # 建立新的cuda版本的软链接
vim ~/.bashrc #将~/.bashrc 或 ~/.zshrc 下与cuda相关的路径都改为 /usr/local/cuda/
export CUDA_HOME=/usr/local/cuda
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda/bin:$PATH
source ~/.bashrc # 发布生效;
nvcc --version #查看当前 cuda 版本