Ubuntu20.04LTS安装Nvidia驱动、CUDA、cuDNN

第一次在CSDN写博客,希望大家给个支持!


前言

最近我一直忙于毕设,在做毕设的过程中,我发现网上的很多安装教程比较陈旧,不适用于新版本,所以我将我安装过程中的一些步骤和细节的地方记录下来分享给各位和我一样的小白,希望可以让大家节省安装软件的时间。


参考文章:
https://www.jianshu.com/p/59b0cbdaa38c
https://blog.csdn.net/weixin_43087913/article/details/109305574
https://zhuanlan.zhihu.com/p/72298520

一、安装Nvidia驱动

Ubuntu系统默认没有Nvidia官方驱动,所以必须要安装。Nvidia驱动可以说是我安装过程中最头疼的一个,踩了很多坑。

1.关闭secure boot

一定要禁用BIOS中的secure boot,因为此方法使用第三方源安装显卡驱动,不禁止secure boot会导致安装的驱动不能使用,禁用也不会有多大安全隐患。我就因为一开始没有禁用security boot,导致安装失败,浪费了很多时间。

2.禁用nouveau

创建下面文件:
sudo vim /etc/modprobe.d/blacklist-nouveau.conf

在文件内插入以下内容():
在VIM中,输入i即可插入文字,鼠标点击右键可以看到粘贴选项,输入完成后按esc退出插入模式然后再输入wq保存并退出VIM回到Terminal

blacklist nouveau
options nouveau modeset=0

执行以下命令使禁用生效并且重启:

sudo update-initramfs -u
sudo reboot

重启之后验证是否生效:

lsmod | grep nouveau

如果没有输出,则禁用成功。

3.禁用图形界面

在进行这一步之前,最好先把驱动下载到主目录下,否则关闭图形界面后无法显示中文字符,很难CD到下载目录下进行安装。
安装Nvidia驱动和CUDA之前,务必确保关闭图形界面!
对于18.04及以上版本来说,采用以下命令:
关闭用户图形界面 :

sudo systemctl set-default multi-user.target
sudo reboot

打开图形界面:

sudo systemctl set-default graphical.target
sudo reboot

对于一些低版本来说则是:
关闭图形界面:

sudo service lightdm stop 

打开图形界面:

sudo service lightdm start 

然后按Ctrl-Alt+F7即可恢复到图形界面。

关闭图形界面重启后,系统并不会自动加载命令行界面,按 Ctrl-Alt+F1 进入命令行界面,输入用户名和密码登录。在纯命令行下操作需要我们掌握一些基本命令,不会的直接百度好了,这里就不讲了。

4.手动安装显卡驱动

首先完全卸载以前安装的显卡驱动:

如果以前是通过ppa源安装的,可以通过下面命令卸载:

 sudo apt-get remove --purge nvidia*

如果以前是通过runfile安装的,可以通过下面命令卸载:

sudo ./NVIDIA-Linux-x86_64-384.59.run --uninstall

在官网下载合适的驱动,然后给驱动添加权限,然后运行,对于新版本来说,-no-opengl-files 指令可能无效,需在安装节目显示后在options菜单中禁用openGL组件安装

sudo chmod u+x NVIDIA-Linux-x86_64-390.87.run 
sudo ./NVIDIA-Linux-x86_64-390.87.run –no-opengl-files

在安装过程中的一些选项我没有都试过,这里就不讲了,大家可以尝试一下,我记得第一个我选了NO,其他的有些不记得了。安装成功后重启. 用nvidia-smi指令试一下, 如果看到类似下图, 恭喜你, 驱动安装成功. 或者看到附加驱动显示继续使用手动安装的驱动.
在这里插入图片描述
打开图形界面,重启,安装之后在软件和更新当中会显示如下图:
在这里插入图片描述
至此,Nvidia驱动安装成功

二、安装CUDA

CUDA(ComputeUnified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。

1.安装必备库

为cuda安装必备库

sudo apt-get install freeglut3-dev libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

可能需要一些时间,耐心点。

2.下载CUDA

在这里插入图片描述
将下载好的包移动到主目录下,关闭图形界面并重启
安装时候依旧要加-no-opengl-files参数, 之后一路默认就好. 最好不要安装与OpenGL相关的,不要照搬代码,看看你自己是什么版本和文件名

sudo sh cuda_10.0.130_410.48_linux.run –no-opengl-files

然后会看到三个installed。添加环境变量:

vim ~/.bashrc

在最后追加:

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

保存并退出后:

source ~/.bashrc

使修改生效,
验证安装成功,

cat /proc/driver/nvidia/version
nvcc -V

如果安装成功,会显示对应版本信息,这里我就不放图了,然后就可以打开图形界面重启了。网上有一些让你make然后运行官方示例验证的,自己找一下吧,我觉得没啥必要。

三.安装cuDNN

cuDNN是用于深度神经网络的GPU加速库。

1.下载对应版本cuDNN

到官网下载,需要账号登录,初次下载可能还要你填个问卷,可能是最近大环境不好,外网访问很吃力,我第一次下花了好几个小时时间,一直卡在登陆界面上不去。耐心点多试几次。选择合适的版本下载(cuDNN Library for Linux)即可。

2.覆盖CUDAs

然后解压,并进入到相应目录,运行以下命令:

sudo cp cuda/include/cudnn.h /usr/local/cuda-10.1/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64
sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h 
sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*在这里插入代码片

以配置cuDNN环境。通过:

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

有些旧版本这里不太一样:

在这里插入代码片

查看版本信息,如果能看到版本信息,则证明cuDNN安装成功。
在这里插入图片描述
至此全部完工。

总结

配置环境对我这样的小白来说还是比较难的,希望大家不要灰心,多Google,百度也可以,网上一些教程可能跟最新版本有所冲突,大家多查查资料,应该都能配置成功。如果有错误的地方,希望大家及时指正。
最后,希望大家点赞关注,写博客不易,给点支持吧亲QAQ!

  • 2
    点赞
  • 21
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值