Ubuntu环境配置caffe开发环境(cuda cudnn + caffe 调试 详细版)

【摘要】本文主要介绍了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等库,遇到相关报错时会知道该命令的好处


3. cuda8.0的安装

    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终端),执行以下指令:

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值