Ubuntu 16.04下 cuda 8.0+cuDNN 5.1 安装

cuda是NVIDIA的编程语言平台，想使用GPU就必须要使用cuda。

cuDNN是GPU加速计算深层神经网络的库。

环境：Ubuntu 16.04 64bit

            Nvidia GeForce GTX 960M （驱动Nvidia-Linux-x86_64-384.130）

安装：cuda 8.0.44

            cuDNN 5.1

显卡驱动安装：https://blog.csdn.net/feishicheng/article/details/70662094

链接：https://pan.baidu.com/s/1dAv_HKIWy6Kk5kwrIIza7A密码：olku

下载CUDA

去官网https://developer.nvidia.com/cuda-downloads下载CUDA。

我选择的是：

Operating System : Linux

Architecture : x86_64

Distribution : Ubuntu

Version : 16.04

Installer Type : runfile(local)

安装CUDA

安装：

下载后即可开始安装CUDA，这里说明一点，在安装CUDA时也可以安装NVIDIA驱动，但是一般选择先安装NVIDIA驱动，我安装的是NVIDIA 384版本。

执行以下命令：

sudo bash cuda_8.0.44_linux.run

安装过程中只需要注意在出现
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 361.62
选择否即可，其余的都直接默认或者选择是。

配置环境变量： 

执行命令

sudo vim ~/.bashrc

如果提示没有安装vim，则执行以下命令安装vim

sudo apt-get install vim

安装完后，可以使用以下命令查看

update-alternatives--display vi #已经添加到系统

vim --version                    #查看版本                           

进入.bashrc后，在末尾处添加

export PATH="/usr/local/cuda-8.0/bin:$PATH"

export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:&LD_LIBRARY_PATH"

测试CUDA的Samples ：

执行以下命令

cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery

sudo make

./deviceQuery

如果显示的是一些关于GPU的信息，则说明安装成功了。

下载cuDNN

在官网下载cuDNN5.1。https://developer.nvidia.com/rdp/cudnn-archive（注意CUDA与cuDNN相对应的版本）

安装cuDNN

下载完成后，进入相应的目录，解压文件。

确保cuda的安装目录是/usr/local/cuda/，之前安装cuda时默认安装目录是这个，然后执行以下命令：

sudo cp cuda/include/cudnn.h /usr/local/cuda/include

sudo cpcuda/lib64/libcudnn* /usr/local/cuda/lib64

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

执行完上述命令后，如果是系统中没有其他版本的cuDNN，则安装完毕，可以执行以下命令查看cuDNN的版本：

1sudo cat /usr/local/cuda/include/cudnn.h |grep CUDNN_MAJOR-A2

最后

以上就是忧伤可乐为你收集整理的Ubuntu 16.04下 cuda 8.0+cuDNN 5.1 安装的全部内容，希望文章能够帮你解决Ubuntu 16.04下 cuda 8.0+cuDNN 5.1 安装所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。