基于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