Ubuntu16.04安装cuda9.0+cudnn7.0

Ubuntu16.04安装cuda9.0+cudnn7.0

这篇记录拖了好久,估计是去年6月份就已经安装过几遍,然后一方面因为俺比较懒,一方面后面没有经常在自己电脑上跑算法,比较少装cuda和cudnn。但是最近课余时间还行,索性一起整理出来,方便以后查看。

检查自己的计算机是否具备CUDA安装条件

  • 检查GPU是否支持CUDA

    lspci | grep -i nvidia
    

    显示出NVIDIA GPU版本信息

  • 去CUDA的官网查看自己的GPU版本是否在CUDA的支持列表中(https://developer.nvidia.com/cuda-gpus

    uname -m && cat /etc/*release
    

  • 验证是否安装了gcc

    gcc --version
    

    如果没有,那就安装之

    sudo apt install gcc
    
  • 验证系统是否安装kernel header和 package development

    uname -r
    sudo apt install linux-headers-$(uname -r)
    

    满足要求后,便可进行下面的正式安装过程。

    注意:CUDA提供两种安装方式:package manager安装和runfile安装。

    这里我选择runfile安装.。采用runfile安装,CUDA自带的驱动可能无法定位内核信息,所以我们先安装NVIIDA驱动。

安装NVIDIA驱动

  • 事先有装过但失败的,先卸载

    sudo apt remove --purge nvidia*
    
  • 禁用nouveau驱动

    lsmod | grep nouveau
    

    有输出,则代表nouveau正在运行, 需要我们手动禁掉nouveau。

    在Ubuntu16.04中,我们可以通过以下操作禁用nouveau:

    1).在/etc/modprobe.d中创建文件blacklist-nouveau.conf

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

    在文件中输入以下内容:

    blacklist nouveau
    options nouveau modeset=0
    

    2)执行

    sudo update-initramfs -u
    

    3)查看nouveau是否已经成功禁用

    lsmod | grep nouveau
    

    注意:若无内容输出,则禁用成功。若仍有内容输出,请检查上述操作,并重复。操作无误但仍有输出,则可重启,进入用户登录界面时(不要登录进桌面),按Ctrl+Alt+F1进入字符终端界面,输入用户名和密码进行登录,登录成功后,再次查看

    这时,我们应该已经成功禁用了nouveau,无输出

  • 关闭图形化界面

    sudo service lightdm stop
    
  • 安装驱动
    NVIDIA驱动安装也有多种方式,runfile文件形式的安装和apt安装

    我用的是apt安装:

    1).添加Graphic Drivers PPA

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

    2).寻找合适的驱动版本

    ubuntu-drivers devices
    

  如上图,我的显卡它建议的是 nvidia-418

  3).安装nvidia driver
 sudo apt install nvidia-418

速度比较慢,应该不是ubuntu源的问题,用了墙外的服务器apt install速度也还是不快,安心等”戴“吧

  4).安装完成后,重启
 sudo reboot
  5).进入桌面,执行下面的命令,查看驱动的安装状态
 sudo nvidia-smi
 sudo nvidia-settings
![](https://img2018.cnblogs.com/blog/1092854/201903/1092854-20190327215611050-860464418.png)

安装CUDA

  • 到NVIDIA官网中,找到CUDA9.0并下载,下载链接

之后,按Ctrl+Alt+F1进入字符终端界面,登录后关闭图形化界面

sudo service lightdm stop

这边我其实用的是命令行直接下载,因为我不喜欢等下再去找文件,而且cuda那个貌似直接下载也有点毒,每次到最后就停在那里了,令人生气,所以可以的话我一般用命令行下载(下面就有一个例子)

wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda_9.0.176_384.81_linux-run

找到下载文件,键入类似命令安装:

sudo sh cuda_9.0.176_384.81_linux-run

会有很长一段,直接ctl+c跳过,出现类似下面的选项,除了第二个选择“是否为NVIDIA安装驱动”,

选择否,不然之前的显卡驱动就白折腾了。

可以按给出的选择

最后会看到cuda驱动、sample、tookit已经安装成功,但缺少一些库。

添加这些库
sudo apt install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev
  重新启动图形化界面
sudo service lightdm start
  同时按Alt + ctrl +F7,返回到图形化登录界面。

如果能够成功登录,则表示不会遇到循环登录的问题,基本说明CUDA的安装成功了

  检查Device Node Verification
ls /dev/nvidia*

  a、若结果显示
/dev/nvidia0    /dev/nvidiactl    /dev/nvidia-uvm
  或显示出类似的信息,应该有三个(包含一个/dev/nvidia-uvm的),则安装成功。

如果只有两个或者一个都没有,文末有记录解决方法。

  设置环境变量
sudo vim ~/.bashrc

注意:.bashrc是当前用户生效的环境变量,(比如我是lzm)如果想对全部用户生效,可以在/etc/profile文件中设置

  按i(inset)进行编辑,在打开的文件末尾,添加以下两行,然后按esc 后按:wq保存退出
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
  保存文件,让环境变量生效
source ~/.bashrc
  检查上述的环境变量是否设置成功。

  a.验证驱动版本
cat /proc/driver/nvidia/version
  b.验证CUDA Toolkit
nvcc -V
  c.最后,尝试编译cuda提供的例子,看cuda能否正常运行

  打开终端输入
 cd /home/lzm(user_name)/NVIDIA_CUDA-9.0_Samples
 make

系统就会自动进入到编译过程,整个过程大概需要十几到二十分钟,请耐心等待。如果出现错误的话,系统会立即报错停止。如果编译成功,最后会显示Finished building CUDA samples。运行编译生成的二进制文件,编译后的二进制文件默认存放在NVIDIA_CUDA-9.0_Samples/bin中。

  接着在终端中输入 :
 cd bin/x86_64/linux/release
 ./deviceQuery  

  结果如下图所示,其中 Result = PASS代表成功,CUDA安装且配置成功,若失败 Result = FAIL

  最后检查一下系统和CUDA-Capable device的连接情况
./bandwidthTest

  看到显示Result=PASS则成功

![](https://img2018.cnblogs.com/blog/1092854/201903/1092854-20190327220028933-368340141.png)

安装CUDNN

cuDNN的安装是建立在我们成功安装cuda的基础上的,本来我是想用终端直接下载cudnn的像上面的cuda一样,但是当我wget的时候发现报错了:(HTTP request sent, awaiting response… 404 Not Found) 才想起来cuDNN是需要登陆NVIDIA账号才能下载的。

官网下载连接

这里我以tgz文件的安装为例

Attention:

(这边有个要注意的,因为我之前下载过的是tgz压缩包,直接按下面命令解压就行。然后今天下载是以solitairetheme8结尾的,然后刚开始我不知道linux是用什么命令解压这种文件的,所以就去查了一下,就是直接重命名这玩意然后按下面命令解压就🆗了)

  • 解压tgz文件
   tar -xzvf cudnn-9.0-linux-x64-v7.tgz
  • 复制文件到cuda安装路径下
   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 /usr/local/cuda/lib64/libcudnn*
  • 然后查看cudnn版本,出现如下图则安装成功
   cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

Bingo!!!好了,记录一遍也好累啊,真想再重新验证一遍看看是不是完美兼容。可是我有点懒,不想去开其他的机子了,只测试了Ubuntu16.04的可以安装,其他的版本不晓得,不过基本思想是一样的

可能出现的问题

ls /dev/nvidia*

可能出现

ls: cannot access/dev/nvidia*: No such file or directory

或者只出现

/dev/nvidia0    /dev/nvidiactl

选项中的一个或两个,但没有/dev/nvidia-uvm,即文件显示不全。

解决方法:
首先要添加一个启动脚本,如果是第一次打开这个文件,它应该是空的(除了一行又一行的#注释项外)。这文件的第一行是

#!/bin/sh -e
把-e去掉(这步很重要,否则它不会加载这文本的内容),然后把下列内容复制到exit 0之前并保存退出。
/sbin/modprobe nvidia

if [ "$?" -eq 0]; then
# Count the number of NVIDIA controllers found.
NVDEVS=`lspci | grep -i NVIDIA`
N3D=`echo "$NVDEVS" | grep "3D controller" | wc -l`
NVGA=`echo "$NVDEVS" | grep "VGA compatible controller" | wc -l`

N=`expr $N3D + $NVGA - 1`
for i in `seq 0 $N`; do
mknod -m 666 /dev/nvidia$i c 195 $i
done

mknod -m 666 /dev/nvidiactl c 195 255

else
exit 1
fi

/sbin/modprobe nvidia-uvm

if [ "$?" -eq 0 ]; then
# Find out the major device number used by the nvidia-uvm driver
D=`grep nvidia-uvm /proc/devices | awk '{print $1}'`

mknod -m 666 /dev/nvidia-uvm c $D 0
else
exit 1
fi

再次查看

ls /dev/nvidia*

结果显示:

其中三个包括 /dev/nvidia-uvm即可。

posted @ 2019-03-27 22:03  路明  阅读(1064)  评论(0编辑  收藏  举报