基于Ubuntu16.04下CUDA8.0开发环境
admin 于 2018年03月25日 发表在 机器学习笔记

关于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
注意:本站所有文章除特别说明外,均为原创,转载请务必以超链接方式并注明作者出处。 标签:机器学习,cuda