ubuntu20.04安装cuda库

ubuntu20.04安装cuda库

官网安装参考
参考链接：https://blog.csdn.net/h3c4lenovo/article/details/119003405.
参考连接：https://blog.csdn.net/kunhe0512/article/details/125061911
参考连接：https://blog.csdn.net/qq_29600137/article/details/122414416（详细，包括安装顺序，推荐）

1、查看电脑是否有GPU（nivida品牌）

lspci | grep -i nvidia

2、禁用系统自带的集成Nouveau驱动

参考链接：https://blog.csdn.net/xue_csdn/article/details/102391145.

3、查看电脑可以安装的版本

ubuntu-drivers devices

4、安装cuda驱动，选择上述图片recommend的版本

sudo apt install nvidia-driver-XXX
reboot

5、查看显卡信息

nvidia-smi

安装nvidia驱动还有一种方法，就是下载驱动源码安装，便于在图形界面启动不了的时候，使用命令行窗口重装驱动，推荐用这种方法

参考链接：https://zhuanlan.zhihu.com/p/59618999.

6、安装cuda库

Ubuntu 20.04默认g++9版本太高，会导致CUDA无法安装，因此要先降低g++版本到7，将下列代码逐个执行

sudo apt-get install gcc-7 g++-7
 
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 9
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 1
 
sudo update-alternatives --display gcc     //查看默认版本
 
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-7 9
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-9 1
 
sudo update-alternatives --display g++

7、到nivida官网下载自己可安装的版本（向下兼容）

选择适合Ubuntu20.04的cuda版本

终端执行下列代码，下载脚本（脚本下载在当前目录下，故选择合适的位置）

wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux.run  //选择自己的版本

运行脚本进行安装cuda库

//cuda_11.4.0_470.42.01_linux.run，表示为cuda_cuda版本号_显卡驱动最低要求版本号_操作系统名称.run
sudo sh cuda_11.4.0_470.42.01_linux.run

在弹出的安装界面中选“continue”，如果选了会跳出安装，就说明安装失败，给了失败日志的路径，自己查看原因，一般是gcc版本问题，上面提到过降级方法，不赘述。

在这一步输入accept

cuda安装包是自带显卡驱动的，所以这一步按空格去掉安装显卡驱动的选项，然后选择install
成功后需要配置一下环境变量，这步很关键。一开始我忽略了这步，一直以为CUDA安装好了，但是TF运行的时候怎么都找不到GPU。

sudo gedit ~/.bashrc

最后一行增加

$ export PATH=/usr/local/cuda-11.4/bin${PATH:+:${PATH}}
$ export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64\
                         ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

特别注意文件夹路径，有时候文件夹直接是/usr/local/cuda，有时候是别的版本号，要确保文件夹存在。

source ~/.bashrc

更新环境变量。

8、检查是否安装成功

nvcc -V