【深度学习】Ubuntu18.04/20.04安装英伟达Nvidia驱动+CUDA10+CUDNN+软连接环境配置教程

深度学习开发环境配置教程

OS:Ubuntu18.04/20.04

本文主要内容有:Nvidia 显卡驱动安装,CUDA10配置,cuDNN配置

本教程全部步骤经过多次实践踩坑,确认无误,在此分享,希望大家都能少踩坑少走弯路!

一、安装Nvidia显卡驱动

1.  准备显卡驱动

NVIDIA官网下载对应的显卡驱动程序,链接地址可以自动判断你的计算机适配的显卡,一路单击“搜索-DOWNLOAD”下载即可。

Ubuntu系统下的驱动文件为.run文件,文件名如:NVIDIA-Linux-x86_64-3xx.xx.run形式

查询GPU型号方法:终端输入

nvidia-smi

或者输入:

nvidia-settings

2. 禁用开源nouveau驱动

安装vim,在终端输入:

sudo apt-get update
sudo apt-get install vim

编辑 /etc/modprobe.d/blacklist.conf文件:

sudo vim /etc/modprobe.d/blacklist.conf

按键盘上的字母 i 进入编辑模式,在最后添加下面两行代码:

blacklist nouveau
options nouveau modeset=0

按Esc键退出编辑模式,再输入“ :wq ”,回车,即可保存退出。

然后输入:

sudo ldconfig

再执行下面的命令,使配置生效:

sudo update-initramfs -u

验证是否成功禁用nouveau,关掉命令行终端,重新打开,然后输入命令:

sudo lspci | grep nouveau

3. 安装显卡驱动

终端更新内核命令:

sudo apt update  
sudo apt install gcc g++ make

好了以后重启电脑。

将前面nvidia 官网下载到的对应显卡版本的驱动(下面XXX是版本的意思) 放置到主文件夹(Home目录)下
举例:NVIDIA-Linux-x86_64-XXX.run (384或者440都可以版本里面的)

按Ctrl + Alt + F3到控制台,输入用户名和密码登陆后,执行关闭当前图形环境命令:

sudo service lightdm stop
sudo telinit 3

注意,再次运行下面命令可以返回命令行:

sudo telinit 3

安装驱动程序,输入下行进入到驱动所在文件夹:

cd /home/用户名

输入:

sudo chmod a+x NVIDIA-Linux-x86_64-xxx.run
sudo sh NVIDIA-Linux-x86_64-xxx.run -no-opengl-files

过程中点击默认选中的按钮,无需更改。

输入以下命令,重启图形界面:

sudo service lightdm start

最后重新启动reboot。

二、安装CUDA环境

1. 准备CUDA文件

官网上下载对应的CUDA安装文件,建议选择run安装。官方地址是,注意此处需要注册成为Nvidia develop 会员。

【深度学习】Ubuntu18.04/20.04安装英伟达Nvidia驱动+CUDA10+CUDNN+软连接环境配置教程

自己根据实际情况选择相应的文件。

2. 安装CUDA

下载完成后,打开终端切换到安装文件所在的目录,执行:

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

这里会出现More(0%),意思是你要阅读这些协议,连续按空格键可跳过,或者按Ctrl+C跳过
之后,按照提示安装即可。依次选择:

accept #同意安装
n #不安装Driver,因为上一步已独立安装显卡驱动
y #安装CUDA Toolkit
<Enter> #安装到默认目录
y #创建安装目录的软链接
n #不复制Samples,因为在安装目录下有/samples

3. 配置环境变量

终端输入:

sudo gedit ~/.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

4. 测试CUDA

测试是否安装成功:

cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery

三、安装cuDNN环境

1. 准备cuDNN文件

下载地址:NVIDIA-cuDNN,此处依然需要登录会员才可下载:

【深度学习】Ubuntu18.04/20.04安装英伟达Nvidia驱动+CUDA10+CUDNN+软连接环境配置教程

本教程在此选择的版本是与CUDA10相对应的cuDNN7.6版本。下载完成后,切换到下载目录解压缩cuDNN包:

sudo tar -zxvf ./cudnn-10.0-linux-x64-v7.6.5.32.tgz

分别执行以下复制文件命令:

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/    # 复制头文件
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/         # 复制动态链接库
sudo chmod a+r /usr/local/cuda/include/cudnn.h            #修改头文件权限值
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*              #修改动态链接库权限值

2. 查看cuDNN版本

在终端命令行输入如下命令确认cuDNN版本:

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

3. 修改软连接

在终端命令行分别输入如下命令进行软连接:

sudo cp /usr/local/cuda-10.0/lib64/libcudart.so.10.0 /usr/local/lib/libcudart.so.10.0 && sudo ldconfig
sudo cp /usr/local/cuda-10.0/lib64/libcublas.so.10.0 /usr/local/lib/libcublas.so.10.0 && sudo ldconfig
sudo cp /usr/local/cuda-10.0/lib64/libcurand.so.10.0 /usr/local/lib/libcurand.so.10.0 && sudo ldconfig
sudo cp /usr/local/cuda-10.0/lib64/libcudnn.so.7 /usr/local/lib/libcudnn.so.7 && sudo ldconfig
sudo cp /usr/local/cuda-10.0/lib64/libcusolver.so.10.0 /usr/local/lib/libcusolver.so.10.0 && sudo ldconfig
sudo cp /usr/local/cuda-10.0/lib64/libcufft.so.10.0 /usr/local/lib/libcufft.so.10.0 && sudo ldconfig

4. 查看环境

在终端命令行通过如下命令显示GPU相关信息,如CUDA版本等,确认安装成功:

nvcc --version

至此全部结束

1、小组所有资源均可通过每日签到任务积分进行免费兑换,记得每日来小组签到哦~ 2、小组资源大多存储在云盘,如出现链接失效请于资源评论区提交链接失效反馈,我们会第一时间修复。 3、所有资源仅供学习交流使用。未经原版权作者许可,禁止用于任何商业环境,否则后果自负。为尊重作者版权,请购买正版作品。 4、所有资源来源于网络,仅限购买正版前临时了解,请下载后24小时内自行删除。 如有侵权,请联系删除(邮箱:hjffily(at)gmail.com 请替换成@)。 5、本站提供的源码、模板、书单等其他资源,暂不包含技术服务,请大家谅解。如有定制需要,请邮件告知具体需求(邮箱:hjffily(at)gmail.com 请替换成@)!
第一学习小组资源共享网 » 【深度学习】Ubuntu18.04/20.04安装英伟达Nvidia驱动+CUDA10+CUDNN+软连接环境配置教程

发表评论

提供最优质的资源集合

立即查看 了解详情