Ubuntu16.04下安装cuda和cudnn(亲测有效)

上一篇博客讲述了Ubuntu16.04下如何安装显卡驱动，这篇继续讲述如何安装cuda和cudnn。首先安装之前首先要确认你需要安装的cuda和cudnn版本

cuda的安装

1.下载cuda安装文件。按照你的系统配置选择安装包，例如我安装的是cuda9.0，那么选择Linux，x86_64，Ubuntu，16.04，最后一个Installer type 选择runfile(local)或者deb(local)都可以，我选择的是deb(local)，下面会分别给出相应的安装过程。
2.1 deb(local)安装
进入到deb(local)位置，我的是放在Download路径下，在此路径下打开终端，输入

1
2
3

sudo dpkg -I cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-get update
sudo apt-get install -y cuda

2.2 安装完毕之后，设置环境变量。打开bashrc文件，将cuda路径写入

1	sudo vim ~/.bashrc

按i进行编辑，把以下3个路径粘贴到bashrc文件中

1
2
3

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
export PATH=$PATH:/usr/local/cuda-9.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0

然后友情提示：输入 :wq,回车即可完成添加路径。
终端运行： source ~/.bashrc
查看cuda 版本
输入 nvcc –-version
3.1 runfile(local)安装
进入到runfile(local)位置，我的是放在Download路径下，在此路径下打开终端，输入

1	sudo sh cuda_9.0.176_384.81_linux.run

然后找个按住回车键不撒手，直到服务条款显示到100%，然后按照下面进行选择：accept，n(不要安装driver)，y，y，y，
安装完成后，设置环境变量，方法同deb(local)安装

cudnn的安装

1.下载对应的安装文件
2.安装cudnn
解压刚下载的文件，出现cuda文件夹，当前路径(我的路径为Download)打开终端

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*

3.查看cudnn版本
终端输入

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

看到红白相间的版本信息，即安装成功。

cuda的安装

cudnn的安装

安装完成，快去感受一下你的算力吧