ubuntu下caffe环境搭建和使用流程

什么是Caffe?

caffe(ConvolutionalArchitecture for Fast Feature Embedding)是由伯克利视觉和学习中心(BVLC)开发,基于C++/MATLAB/Python实现的卷积神经网络框架,作者贾杨清。
它能完成的任务只是卷积神经网络,其他的神经网络RNN等,它是没有能力实现的。另外,caffe是主要用来做计算机视觉方向的。

内部提供了一套基本的编程框架,或者说一个模板框架,用以实现GPU并行架构下的深度卷积神经网络,Deep Learning等算法,我们可以按照框架定义各种各样的卷积神经网络的结构,并且可以再此框架下增加自己的代码,设计新的算法,该框架的一个问题就是,只能够使用卷积网络,所有框架都是再基于卷积神经网路的模型上进行的。

与caffe类似的框架还有google的TensorFlow,facebook的pytorch。

虽然TensorFlow和pytorch的功能更加强大,但是caffe有配置方便、不用编译,灵活易用等特点,广受欢迎,开源项目多、社区活跃,适合快速应用。

为什么在linux下使用?

1.caffe最初只在linux上提供服务,大量开源项目网络都是在linux完成,无法运行在windows版caffe,经常使用的第三方模块在windows下会出现各种难以解决的问题,所以一般的生产力环境大多使caffe工作在linux系统下。

2.linux下caffe使用配置更加方便,不需要windows系统下复杂的环境配置。

caffe安装教程

1.配置要求

系统要求:Ubuntu 16.04

ububtu 是最常见的个人桌面linux系统,具有较好的图形界面能力和使用体验,计算机视觉工作一般都以ubuntu系统作为载体。

ubuntu.png

硬件要求: intel处理器和nvida显卡

深度学习任务对计算机性能要求较高,因为系统主要使用显卡进行cuda计算,因此对CPU的要求并不是很高,但需要较强性能的显卡进行复杂的单精度运算
通常神经网络需要大量显存和内存资源,因此需要8GB以上显存和显存两至三倍的内存才能够胜任计算机视觉任务,一般选择GTX1080以上配置。


显存.png

硬盘:能够装下大量训练数据即可。

电源:保证长时间不断电

2.安装caffe

卷积神经网络需要cuda计算,和cudnn加速,需要先完成驱动安装、cuda安装、cudnn安装

1.显卡驱动安装

1.首先卸载旧版本显卡驱动:

$sudo apt-get purge --remove nvidia* 命令卸载旧版本

2.添加驱动源

$sudo add-apt-repository ppa:graphics-drivers/ppa
$sudo apt update

3.安装驱动
首先,检测你的NVIDIA图形卡和推荐的驱动程序的模型。执行命令:

ubuntu-drivers devices 

输出结果为:

== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001B81sv000019DAsd00001434bc03sc00i00
vendor   : NVIDIA Corporation
model    : GP104 [GeForce GTX 1080 Ti]
driver   : nvidia-driver-396 - third-party free
driver   : nvidia-driver-390 - third-party free
driver   : nvidia-driver-415 - third-party free recommended
driver   : nvidia-driver-410 - third-party free
driver   : xserver-xorg-video-nouveau - distro free builtin

从中可以看到,这里有一个设备是GTX 1080 ,对应的驱动是nvidia-driver-396,415,410 ,而推荐是安装415版本的驱动

$sudo apt install nvidia-driver-415 此处需上nvida官网查询最新驱动版本

OK 驱动安装完成,重新启动,查看系统配置---》详细信息---》关于 ,图形处理是否对应于自己的独立显卡。
在终端输入

nvidia-smi

显示如下显卡信息即完成安装

GPU.png

2.安装CUDA(针对18.04)

cuda是nvida提供的的科学计算
1.查阅 NVIDIA CUDA Installation Guide for Linux
英伟达CUDA安装指南(Linux)
对于后续的学习与工作大有裨益,包含了CUDA的详尽安装说明。当然如果你不想阅读英文,只想尽快安装CUDA,可以参照后续步骤。

2.前往官网下载对应系统版本安装

下载安装.png

3.进入下载文件所在文件夹,在终端输入

$sudo sh cuda_8.0.61_375.26_linux.run 这是文件名

若无法安装,下载deb(local),输入

$sudo dpkg -i cuda-repo-ubuntu1704-9-0-local_9.0.176-1_amd64.deb 这是文件名
$sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
$sudo apt-get update
$sudo apt-get install cuda

看完版权声明后完成安装

4.添加环境变量

$gedit ~/.bashrc

在打开的文件中输入

$export PATH="$PATH:/usr/local/cuda-8.0/bin"
$export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64"

保存并关闭文件后,输入

source ~/.bashrc

完成更改
切换到目录~/ NVIDIA_CUDA-8.0_Samples
输入

make

等待完成后即可
输入

$sudo ./deviceQuery

显示显卡相关信息即成功。

3.安装cudnn

cudnn是nvida提供的神经网络加速工具
1.下载cuDNN,下载之前需要注册Nvidia开发者帐号,按要求注册即可

2.选择cuda对应版本,下载cudnn

3.将其解压后链接至cuda include目录即可

记得改成自己下载的文件名
sudo tar -zxvf ./cudnn-8.0-linux-x64-v5.1.tgz 
cd cuda; sudo cp lib64/lib* /usr/local/cuda/lib64/; 
sudo cp include/cudnn.h /usr/local/cuda/include/  
更新软连接: cd /usr/local/cuda/lib64/
sudo chmod +r libcudnn.so.5.1.10
sudo ln -sf libcudnn.so.5.1.10 libcudnn.so.5
sudo ln -sf libcudnn.so.5 libcudnn.so
sudo ldconfig

4.python安装和caffe安装

不使用anaconda安装caffe非常繁琐,现在已经被放弃,故采取anaconda安装python和caffe
python安装可参考我的 anaconda安装教程
python安装完成后
创建一个环境 或在已有环境运行。

$conda create -n name python=3.6 自行指定python版本

进入环境

$source activate name

输入

$conda install caffe -gpu

等待下载完成
因为anaconda会自动设置,就免去了caffe的繁琐配置
caffe安装完成。
在python中就可以调用caffe的库了

import caffe

如果没有nvida显卡,使用GPU版本,只需要完成上述python安装和caffe安装即可,但是cpu进行深度学习速度非常慢

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,530评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 86,403评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,120评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,770评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,758评论 5 367
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,649评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,021评论 3 398
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,675评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,931评论 1 299
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,659评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,751评论 1 330
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,410评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,004评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,969评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,203评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,042评论 2 350
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,493评论 2 343

推荐阅读更多精彩内容