参考了以下3篇文章:
Ubuntu16.04 +cuda8.0+cudnn+opencv+caffe+theano+tensorflow配置明细
Ubuntu16.04 Caffe 安装步骤记录(超详尽)
ubuntu16.04下caffe安装(GPU版)
主要参考了第一篇,其余两篇用作细节辅助,第一篇原文有些地方有细节错误,有的地方不够明确,这里从头整理下来。
环境:Ubuntu16.04
工具:cuda_8.0.61_375.26_linux.run + cudnn-8.0-linux-x64-v5.1.tgz + opencv-3.1.0.zip
第一步:安装英伟达显卡驱动
首先去官网上查看适合你GPU的驱动,记住对应的版本号,我的是390:
http://www.nvidia.com/Download/index.aspx?lang=en-us
然后在home中执行:
1. sudo add-apt-repository ppa:graphics-drivers/ppa
2. sudo apt-get update
3. sudo apt-get install nvidia-390(390改为你查到的版本号)
4. sudo apt-get install mesa-common-dev
5. sudo apt-get install freeglut3-dev
6. reboot
重启后,输入:
nvidia-smi
如果出现了你的GPU列表,则说明驱动安装成功了。另外也可以通过输入
nvidia-settings
出现以下画面,则表示成功:
* 第二步:安装和配置cuda 8.0*
进入官网下载对应版本,https://developer.nvidia.com/cuda-downloads
我将安装包下载到home目录,然后在终端进入home目录,输入:
1.sudo dpkg -i cuda-repo-ubuntu1604-8-0-rc_8.0.27-1_amd64.deb
2.sudo apt-get update
3.sudo apt-get install cuda
ubuntu的gcc编译器是5.4.0,然而cuda8.0不支持5.0以上的编译器,因此需要降级,把编译器版本降到4.9:
1.sudo apt-get install gcc-4.9 gcc-5 g++-4.9 g++-5
2.sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20
3.sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10
4.sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.9 20
5.sudo update-alternatives --install /usr/bin/g++ g+