基于Ubuntu16.04下CUDA8.0开发环境

关于cuda的下载方式，参考博文《Win7下cuda8.0深度学习开发平台》，此处不在赘述。

本文开发环境：****ubuntu16.04.4-x64 + nvidia-384 + cuda8.0.44 + cudnn-8.0-linux-x64-v6.0

1. 下载cuda，访问官网

官方下载所需cuda的版本，本文选择针对ubuntu16.04的deb安装包，如下：

**2. 安装显卡驱动 **

在安装cuda和cudnn前，需要安装nvidia官网的闭源驱动，可通过如下命令安装（目前最新）:

sudo apt-get install nvidia-384

3. 安装cuda软件包

参考以下步骤依序操作：

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb 
sudo apt-get update
sudo apt-get install cuda

安装过程中，会出现如下打印信息：

4. 添加环境变量

sudo gedit /etc/profile

添加内容：

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

5. 测试安装

测试是否安装成功，会输出的版本信息，如下：

nvcc -V

6. 下载cudnn，访问官网

下载对应的cudnn开发包（注：需先注册），如下：

7. 安装cudnn

只需将解压后的cudnn文件复制到cuda8.0对应目录，即安装完成。

sudo cp include/cudnn.h /usr/local/cuda/include
sudo cp lib64/libcudnn.* /usr/local/cuda/lib64

YY分享