Ubuntu 16.04下配置GPU版CUDA和cuDNN

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/kwame211/article/details/83995046

先介绍一下我自己的配置环境

Ubuntu 16.04

Intel i7 CPU

16G内存

GTX980Ti 显卡

1、安装NVIDIA显卡驱动

(1)先在NVIDIA官网上下载对应的驱动程序,可根据自己的GPU的型号下载相应的.run文件

例如NVIDIA-Linux-x86_64-3xx.xx.run形式的文件名

自己电脑的GPU型号查询可在终端输入:


nvidia-smi
查询。大概在中间位置。也可以输入:

nvidia-settings


(2)禁用开源nouveau驱动(非常重要)

a.如果没有装vim,则先装vim编辑器(当然用gedit也可以,在下面也会介绍到)

安装vim,在终端输入


sudo apt-get update
sudo apt-get install vim
接下来 编辑文件/etc/modprobe.d/disable-nouveau.conf
aa.使用vim(关于vim编辑器的使用,百度上有很多),在终端输入


sudo vim /etc/modprobe.d/disable-nouveau.conf
然后按键盘上的字母 i 进入编辑模式,并在最后添加下面两行

blacklist nouveau
options nouveau modeset = 0
按Esc键退出编辑模式,再输入下面的命令,回车,保存退出。
:wq
然后输入

sudo ldconfig
再执行下面的命令,使配置生效。

sudo update-initramfs -u
bb.使用gedit,在终端输入:

sudo gedit /etc/modprobe.d/disable-nouveau.conf
然后同样在末尾加上那两句,然后鼠标点save保存,后续步骤则跟用vim一样
验证是否成功禁用nouveau

关掉终端,重新打开,然后输入命令:


 sudo lspci | grep nouveau
如果没有内容显示,则禁用成功。

b.配置kernel以text模式启动


sudo vim /etc/default/grub
编辑文件/etc/default/grub,修改下面两句(操作方法同步骤a,建议用gedit)sudo gedit /etc/default/grub
GRUB_CMDLINE_LINUX_DEFAULT="quiet splash"
GRUB_CMDLINE_LINUX=""
将上面两句修改为:

GRUB_CMDLINE_LINUX_DEFAULT="quietsplash text"
GRUB_CMDLINE_LINUX="rdblacklist=nouveaunouveau.modeset=0"
保存退出,并更新。

sudo update-grub

c.重启电脑,在登录界面直接按Ctrl+Alt+F1进入控制台
输入用户名和密码登陆后,执行:


sudo service lightdm stop
关闭图形界面(切记,否则英伟达显卡驱动会安装失败),然后cd命令切换到英伟达显卡驱动安装文件NVIDIA-Linux-x86_64-3xx.xx.run所在的目录,执行:
sudo sh NVIDIA-Linux-x86_64-3xx.xx.run
输入以下命令,重启图形界面

sudo service lightdm start
d.结束后再次修改grub文件
将前面修改的第一行改回原来的。第二行可不修改。本人改完是这样的:


GRUB_CMDLINE_LINUX_DEFAULT="quiet splash"
GRUB_CMDLINE_LINUX="rdblacklist=nouveaunouveau.modeset=0"
保存退出后,再次更新

sudo update-grub
2、安装CUDA以及cuDNN

(1)先在官网上下载对应的CUDA安装文件,建议选择run安装,点这里

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

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

sudo sh cuda_8.0.61_375.26_linux.run
这里会出现More(0%),意思是你要阅读这些协议,连续按空格键可跳过,按照提示输入accept,第一个安装的时候,是安装图形驱动的,由于前面已经安装了NVIDIA驱动,所以选择no,后面全选yes,建议选择默认安装路径。
(2)环境变量配置

sudo gedit ~./bashrc
将以下内容写到文件尾部。
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存后,更新文件
source ~/.bashrc
(3)配置cuDNN
首先在官网上(https://developer.nvidia.com/cuda-downloads)下载CUDA:

下载cuDNN5.1之后切换到下载目录进行解压:

sudo tar -zxvf ./cudnn-8.0-linux-x64-v5.1.tgz 
进入cuDNN5.1解压之后的include目录,在命令行进行如下操作:

cd cuda/include
sudo cp cudnn.h /usr/local/cuda/include  #复制头文件
再将进入lib64目录下的动态文件进行复制和链接:

cd ..
cd lib64
sudo cp lib* /usr/local/cuda/lib64/    #复制动态链接库
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.5    #删除原有动态文件
sudo ln -s libcudnn.so.5.1.10 libcudnn.so.5  #生成软衔接
sudo ln -s libcudnn.so.5 libcudnn.so      #生成软链接
sudo ldconfig      #使配置生效
(4)测试CUDA的samples
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
sudo make 
sudo ./deviceQuery
如果显示一些关于GPU的信息,怎说明安装成功。

可通过

nvcc --version
查看CUDA版本。
 

猜你喜欢

转载自blog.csdn.net/kwame211/article/details/83995046