ubuntu18.04安装NVIDIA驱动和cuda及CUDNN

此记录由参考多位博主的记录总结而来,我记录下，看是否对大家有所帮助

首先禁用nouveau

sudo gedit /etc/modprobe.d/blacklist-nouveau.conf

在新建的文件中写入

blacklist nouveau
options nouveau modeset=0

随后使写入文件生效

sudo update-initrams -u
sudo reboot   #重启电脑

重启后查看禁用是否成功

lsmod | grep nouveau

如果执行这句话，没有输出则说明已经禁用成功

随后打开ubuntu自带的软件和更新功能，点击附加驱动，随后就会发现有会搜索出很多的驱动（开源、专用、各种版本号的都有）；去到英伟达驱动的官网，查看自己的电脑显卡支持的版本号范围，随后在这个范围中回到附加驱动页面，选中比较中间的版本号，并点击应用更改即可。
随后终端重启

sudo reboot

重启后输入

nvidia-smi

如果出现GPU列表则安装成功
如果没有出现，尝试先输入以下语句，随后在点击附加驱动的版本号应用更改

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update

---

---

以上内容为安装显卡驱动，安装好以后准备安装对应版本的cuda和CUDNN
首先在终端输入

nvidia-smi

可以查看显卡的详细信息
同时上面有会有推荐的cuda的版本，我这里显示是CUDA Version：10.1
随后去官网查看10.1版本对应的cuda安装包，大概下载是2.4G大小；
下载完以后将其保存在单独的文件夹中，并将这个文件夹用终端打开，打开以后输入以下语句：

sudo sh cuda_10.1.243_418.87.00_linux.run

随后在跳出来的安装界面中，在【X】Driver栏按一次enter键，使X消失；
随后选中Install，再按一词enter键即可进行安装；
等待一会儿，安装完以后，输入以下语句添加环境变量

vi ~/.bashrc

再打开的文件中，添加以下语句

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

使环境变量生效

source ~/.bashrc

在终端输入以下语句，最后出现pass即可视为安装成功

cd /usr/local/cuda-10.1/samples/1-Utilities/deviceQuery
sudo make
./deviceQuery

---

----------------------------------------------------------------------------------------------------------以下内容为安装对应版本的CUDNN

同样在英伟达的官网下载对应版本的CUDNN，就是下载前需要先注册一个账号
我这里下载的是cudn-10.1-linux-ppc64le-v8.0.4.3.0
随后将下载后的文件夹进行解压，解压后
分别到include和lib对应的文件夹输入以下语句

sudo cp cudnn.h /usr/local/cuda/include/ 
sudo cp cudnn_version.h /usr/local/cuda/include/ 
sudo cp libcudnn* /usr/local/cuda/lib64/ 

随后输入

sudo chmod a+r /usr/local/cuda/include/cudnn.h 
sudo chmod a+r /usr/local/cuda/include/cudnn_version.h 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn* 

以上步骤完成后，即可在终端查看安装的CDNN版本

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2