【摘要】本文主要介绍了ubuntu环境下配置caffe开发的详细过程。本人由于缺乏经验,期间遇到了很多问题,经过了大约一周的时间,终于实现了初步理想效果。故撰此文,希望能带给其他同学有效的帮助。
一、软硬件信息
电脑硬件:如下图所示,本人电脑是属于较低配置,如果要深入学习caffe,训练GoogleNet,VGGNet等大型网络,建议使用更高配置(具体可上网搜索)。
软件信息:
Ubuntu16.04-LTS
cuda_8.0
cudnn-8.0-linux-x64-v6.0
NVIDIA-linux-x86_64-384.98.run
anaconda2
二、详细配置流程
1. ubuntu16的安装
ubuntu16.04-LTS版本下载地址:http://cn.ubuntu.com/download/点击打开链接 建议使用64bit版本,因为caffe的部分编程以及数据结构是基于64bit格式。
此外,由于要用到物理显卡,不建议使用虚拟机(无法检测物理显卡,驱动安装无法成功)。详细安装系统的教程请参考网友分享,笔者由于第一次安装双系统,曾经导致原有的win7也崩溃,在此特别写明部分注意事项:
1) 利用UltraISO软件把ubuntu.iso文件制作一个启动盘,在window系统下划出一块空闲磁盘空间给ubuntu。然后重启电脑,进入boost模式(跟进入BIOS的方式一样,因电脑不同而不同)
2) 按照安装索引完成ubuntu的安装。建议安装中文版ubuntu,笔者第一次安装英文版ubuntu后,无法使用汉语输入法。
3) 其余默认,约二十多min即可完成安装。
接下来就是caffe环境配置啦。首先贴出caffe官网教程,供有兴趣的同学查阅。http://caffe.berkeleyvision.org/install_apt.html 点击打开链接
个人认为,教程准确简洁,对于依赖库和软件包都罗列明了,但由于很多具体步骤没有体现,对于第一次配置环境的朋友来说,容易出现各种各样的问题(说多了都是泪。。。。).
2. 安装通用依赖项
根据教程,执行以下命令:
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev #以上都是通用依赖库
sudo apt-get install libatlas-base-dev #安装BLAS, 一般选择Atlas, 如果选择Openblas or MKL ,配置会复杂一点
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev #安装gflags,glog,Mdb等库,遇到相关报错时会知道该命令的好处
cuda安装是环境配置过程中最坎坷的一环,笔者参考了很多网友的分享,但参差不齐,说法不一,导致重复安装了3次才得以成功。之所以使用cuda8.0,是为了更好地与caffe兼容,下载地址如下:https://developer.nvidia.com/accelerated-computing-toolkit
1)下载NVIDIA驱动
首先去官网(http://www.nvidia.com/Download/index.aspx?lang=en-us)查看适合自己显卡的驱动(下载runfile文件),笔者选择的NVIDIA-linux-x86_64-384.98.run。
2)安装驱动
ubuntu一般自带nouveau驱动,通过如下命令可以获得物理显卡信息。如没有结果显示,说明显卡不可引用,安装会失败
lspci | grep nvidia sudo apt-get install linux-headers-$(uname -r) #为当前系统核安装头文件和开发包
按ctrl+alt+F1进入控制台(就是一片黑屏,类似于cmd终端),执行以下指令: