本文主要是介绍12 张图教你傻瓜式配置caffe (Ubuntu14.04 安装CUDA7.5 + Caffe + cuDNN),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
据说每一个搞深度学习的开始总会在caffe等配置上折腾很久,身边有配一个星期的;两个星期的;一个月的;在网上还看到折腾三个月把电脑都整奔溃的。。网上各种教程五花八门,稍不留心就带进坑!!
我第一次让室友帮我的(不到两小时搞定,有个大神盆友多幸运),这次重装系统,再让他指导我配置caffe。为防止自己遗忘,边安装边制作一个超详细的说明。
配置说明:Ubuntu14.04 cuda7.5 cudnn V4 (如果你是Ubuntu14.04就可以粘贴复制无脑安装了)
一 :安装caffe前安装驱动,网上教程都是下载驱动,下个毛,Ubuntu有自带的!!
在Ubuntu左上角点击输入:add 找到Additional Drivers
点击出现下图:
选NVIDIA的驱动,即试试前两个,我的第二个重启进不去,选第一个tested 可以。
点击 Apply Changes
完成后点击 restart 重启,驱动安装完成!!
二 : 进入caffe官网(看官网,官网最权威) 注:真的想无脑安装可以跳过第二部分进入 三 开始粘贴复制!!
百度:caffe官网
点击进去
看到:
点击Installation instructions
看到:
点击 Ubuntu Installation
看到:
三 :caffe配置 (可以粘贴复制了)
Ctr+Alt+T (打开终端都告诉你了哈!!零基础安装,哈哈。。。)
依次输入1,2句配置caffe依赖项。(刚用Linux的同学用此教程成功安装caffe,他建议我数字1,2等后面加点,OK!)
2. sudo apt-get install --no-install-recommends libboost-all-dev
四 安装cuda7.5
下载cuda7.5点击打开链接
选择深绿色部分 (大神的选择) 点击下载
下载完成会自动跳出一个界面:
(如果以前下载过点击压缩包两下就跳出来,挺逗。。)
点击 右上角的 install
在终端cd 到cuda所在目录 例如我的 cd /home/../Documents
依次输入3 4 5 句 安装cuda
4. sudo apt-get update
5. sudo apt-get install cuda
五 : 设置cuda环境变量
终端输入: sudo gedit ~/.bashrc
跳出一个文档,末尾加下面两句
export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH
注:保存关闭后会在命令行有warning,大神请你无视。
六:配置blas ,Python 依赖库
终端输入第 7 8 9 句
8. sudo apt-get install python-dev
9. sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
以上caffe cuda配置基本结束,命令行输入仅仅9句,网上教程太坑拉。。。。
七:验证cuda的安装
依次 输入10 11句
10. nvidia-smi
结果
11. nvcc --version
结果
不报错就对了!
cd进入/usr/local ,看看有没有出现安装的cuda 和cuda7.5,如图:
有就ok!
八:验证caffe的安装
命令行输入第12句 cd 到下载的caffe目录
12. cd /home/...your dir/Documents/caffe-master
输入第13句并打开复制生成的Makefile.config文档,将 # CPU_ONLY := 1的#去掉
CPU_ONLY := 1
Makefile.config文档如图:
14 make (同学用我的教程建议我加一句话:在命令行输入:make )
然后再加上 # ,在GPU模式下编译,输入15句
# CPU_ONLY := 1
15 make all -j4
cuda installed !!
验证完毕,caffe cuda 配置验证15句话!!
九:cudnn安装 (cudnn下载要在官网注册下载,很麻烦,如果需要留下邮件发给你。)
顺便讲讲cudnn安装,cudnn不必要部分 ,起加速作用
cudnn不需要安装,最简单的方法是把cudnn的lib64拷到/usr/local/cuda/lib64目录下:我的是:
sudo cp /home/hk/Downloads/cuda/lib64/* /usr/local/cuda/lib64
把cudnn的cudnn.h拷到/usr/local/cuda/include目录下,我的是:
sudo cp /home/hk/Downloads/cuda/include/* /usr/local/cuda/include
详细请看我另一篇:点击打开链接
这篇关于12 张图教你傻瓜式配置caffe (Ubuntu14.04 安装CUDA7.5 + Caffe + cuDNN)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!