分享在Ubuntu 14.04下CUDA8.0 + cuDNN v5 + Caffe 安装配置过程。 一、硬件与环境
显卡:GTX 1080
系统:Ubuntu 14.04
CUDA:cuda_8.0.44_linux.run
cuDNN:cudnn-8.0-linux-x64-v5.1.tgz
注意:
GTX1080显卡必须用CUDA 8.0版本。CUDA从此处下载。切记,千万不要下载 deb 包,否则后方无数坑在等着你。
CUDA下载界面
GTX1080显卡必须用cuDNN-8.0-V5.1版本,不然用 caffe 跑模型,用 CPU或GPU显卡跑精度正常,一旦开启cuDNN模式,精度(acc)立刻下降到 0.1 左右,loss 非常大。cuDNN在此处下载。下载需注册。最好注册一个账号,选择对应的版本,不要用网上其他教程给的现成的包,出问题的概率非常大。
二、安装:注意:一定要按顺序!
注意:一定要按顺序!
注意:一定要按顺序!
本文不关注。假设你已经将此系统做过稳定的开发环境,这不是全新安装后的系统,至少 git 等最常用包已经安装。缺什么装什么,apt-get 大法好。
2. 禁用 nouveau 驱动打开终端,输入$ sudo gedit /etc/modprobe.d/blacklist-nouveau.conf,输入
blacklist nouveau options nouveau modset=0保存退出,执行$ sudo update-initramfs -u,禁用结束。重启电脑。
验证驱动是否禁用成功:
输入$ sudo lspci | grep nouveau,如果没有内容,则禁用成功。
什么都没有,则禁用成功
3. 安装CUDA 8.0运行下载好的 run file (假设 cuda_8.0.44_linux.run 在家目录下)。
$ sudo sh cuda_8.0.44_linux.run
一路回车即可。
验证是否安装成功:
输入$ ls /dev/nvidia*,若生成 4 个左右 Nvidia 开头的文件(夹),说明此步安装成功。此时已经安装好显卡驱动和CUDA 8.0。输入$ nvidia-smi可查看显卡驱动和其他信息。
生成了五个 nvidia 开头的文件夹
显卡驱动版本信息
注意:网上其他教程由于年代久远,里面不建议安装此 CUDA 包中的显卡驱动。然而,这个包中的显卡驱动大版本为 367 ,非常新,可以用。
4. 配置环境变量输入$ sudo gedit /etc/profile,打开 gedit ,最后两行输入
export PATH=/usr/local/cuda-8.0/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH保存退出,环境变量配置完成。
5. 安装 cuDNN你已经下载好 cudnn-8.0-linux-x64-v5.1.tgz,并将其放在家目录($ cd ~)下。
按顺序输入以下代码:
注意:libcudnn.so后面跟的数字可能和你下载的 cudnn 包小版本的不同而不同,去~/cuda/lib64下看一眼,相对应地进行修改。
6. 安装 caffe (1) 下载 caffe在家目录执行 $ git clone https://github.com/BVLC/caffe.git,~下生成文件夹 caffe。
(2) 安装依赖 $ sudo apt-get install libatlas-base-dev libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler (3) 编译 caffe打开 caffe 所在目录,找到Makefile.config.example文件,将其改名为Makefile.config,打开。将# USE_CUDNN := 1一行开头的#删除,保存。
在 MakeFile.config 中开启 cuDNN模式
打开终端,输入
cd ~/caffe make -j