ubuntu20.04安装cuda库
参考链接:https://blog.csdn.net/h3c4lenovo/article/details/119003405.
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
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库
wget 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
|