首页 > 技术文章 > Ubuntu14.04安装CUDA8.0与Cudnn5.1

aiuestcer 2017-04-24 11:16 原文

版权声明:本文为博主原创文章,未经博主允许不得转载。

一、安装CUDA

1.参考资料:官方文档 NVIDIA CUDA INSTALLATION GUIDE FOR LINUX

2.安装方式

官方文档中有deb与run两种安装方式,我选择run安装方式

    选择理由:The Runfile installation installs the NVIDIA Driver, CUDA Toolkit, and CUDA Samples
via an interactive text-based interface.一站式安装 NVIDIA Driver, CUDA Toolkit, and CUDA Samples,方便快捷!

3.安装前准备

    跟着官方文档的Pre-installation Actions走就对了,一般不会出现什么问题的!

4.关闭nouveau

 4.1

$sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
           blacklist nouveau
           options nouveau modeset=0

4.2 Regenerate the kernel initramfs:

$sudo update-initramfs -u(这一步很关键,如果跳过直接重启,则nouveau关闭失败,看了官方文档之后才知道了,看官方文档的重要性)

4.3 重启

$sudo reboot

4.4 检验是否关闭nouveau

$lsmod | grep nouveau

    若输入上述命令之后终端没有任何显示,则成功关闭nouveau

5.安装Runfile

5.1 Ctrl +F1+Alt

5.2 输入自己的用户名与密码

5.3 

$sudo service lightdm stop

5.4 安装我所下载的版本,不同版本注意修改

$sudo sh cuda_8.0.61_375.26_ linux.run

5.5 环境变量设置

*官方文档此步为重启,我配置完环境变量之后再重启

$sudo gedit /etc/proflie
           export PATH=/usr/local/cuda-8.0/bin:$PATH
           export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH(输入内容)
$source /etc/profile
$sudo gedit /etc/ld.so.conf.d/cuda.conf 
           /usr/local/cuda/lib64(输入内容)
$sudo ldconfig

5.6 重启

$sudo service lightdm start
$sudo reboot

6.检验安装CUDA是否成功

$nvidia-smi

    如果出现NVIDIA相关信息,则表明安装成功!

二、安装Cudnn

1.参考资料:http://blog.csdn.net/u013645510/article/details/53448806

2.解压

$tar xzvf cudnn-8.0-linux-x64-v5.1.tgz

    我所下载的cudnn版本,我下载在主文件夹里边,直接解压,如果下载到其他文件夹则cd到那个文件夹目录下在解压.

3.安装

$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
$ sudo cp cuda/lib64/libcudnn* /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(不同版本的cudnn,libcudnn.so.5.1.x不同)
$ sudo ln -s libcudnn.so.5 libcudnn.so
$ sudo ldconfig -v

三、小结

    安装CUDA可谓是一波三折,折腾了这么久,总算安装好了,安装CUDA只是基础,后面还得了解它!

    折腾了这么久,也算有收获的,明白了官方文档的重要性,官方文档是了解工具的第一手资料,不要本末倒置,官方文档优先级远远高于技术博客等,技术博客等文档仅供参考解惑之用不能形成依赖,前人经验富丰那也是别人的绝非自己的!

    纸上得来终觉浅,绝知此事要躬行!

    Life is short,save time!

    深度学习之路漫长,需要沉下心来脚踏实地学下去,逐渐扒开深度学习的大门!

 

推荐阅读