ubuntu下的cuda安装

1.先查看驱动

看有没有问题，有问题就先装一下驱动

Ubuntu20.04安装Nvidia显卡驱动教程_ubuntu20.04安装nvidia驱动-CSDN博客

lspci|grep NVIDIA

查看一下ubuntu版本

lsb_release -a

2.对应ubuntu版本下载相应的cuda版本

 根据生成的命令下载cuda包

wget https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux.run
sudo sh cuda_11.0.3_450.51.06_linux.run   #根据自己生成的来

 有的不建议使用第二行官网中的代码，而是使用下方类似NVDIA驱动安装的代码

（本人没感觉有啥不同）

sudo chmod +x cuda_11.0.3_450.51.06_linux.run

sudo ./cuda_11.0.3_450.51.06_linux.run

3.安装

把驱动的选项取消掉，往下走，accept，install 等待一会就好了

有的是在原来基础上装新的，就upgrate all。

 装好显示：

 4.查看cuda

nvcc -V

如果还显示旧的或者有错误，那就更新配置文件：

（1）进入

sudo gedit ~/.bashrc

 （2）在最后一行添加（如果是11.0是这个）

export PATH="/usr/local/cuda-11.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH"

 （3）保存后，输入下面代码令其生效

source ~/.bashrc

（4）再次查看cuda则显示已安装的版本，安装完成

5.如果需要删除cuda 

由于pytorch并没有支持到CUDA11.0，因此卸载并重新安装别的版本。

cd /usr/local/cuda-11.0/bin/
sudo ./cuda-uninstaller
sudo rm -rf /usr/local/cuda-11.0

6.cudnn的安装

参考ubuntu下的cudnn安装-CSDN博客