caffe安装教程(ubuntu14.04+cuda8.0+cudnn5.1+opencv2.4.10)详细

电脑配置:

显卡:GTX 1060(6G)

系统:ubuntu14.04

CUDA: cuda-repo-ubuntu1404-8-0-local_8.0.44-1_amd64.deb

cuDNN: cudnn-8.0-linux-x64-v5.1.solitairetheme8

1.安装系统(ubuntu14.04)

1)做系统盘

工具:rufus-3.1.exe;ubuntu-14.04.1-desktop-amd64.iso;一个u盘。

2)安装系统

https://blog.csdn.net/ezhchai/article/details/52891014

安装报错:https://blog.csdn.net/king_haw/article/details/70753324(报错只是1060对应ubuntu14.04才出现,目测其他版本没有问题。)

我是将1060显卡拆下,换上580(只要不是1060)显卡,然后按照教程安装就能成功;安装后进入ubuntu14.04,安装1060显卡驱动,之后关机换上1060显卡即可。

驱动安装:先去NVIDIA官网查看自己电脑具体驱动版本

https://blog.csdn.net/king_haw/article/details/70143875

安装双显卡的教程:https://blog.csdn.net/masa_fish/article/details/51882183

2.安装CUDA

安装runfiles格式请参考:https://blog.csdn.net/masa_fish/article/details/51882183

百度搜索CUDA8.0,下载cuda-repo-ubuntu1404-8-0-local_8.0.44-1_amd64.deb(不同时间版本不同,这里以8.0.44-1为例)

打开终端(ctrl+alt+t)输入你下载的CUDA路径,如:

cd /home/myubuntu/download

然后依次输入以下命令(下载界面也有):

sudo dpkg -i cuda-repo-ubuntu1404-8-0-local_8.0.44-1_amd64.deb

sudo apt-get update

sudo apt-get install cuda 

设置环境变量:打开.bashrc文件

    sudo gedit  ~/.bashrc

在文件末尾添加:

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

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

保存然后在终端运行:

source ~/.bashrc

  检查:在终端输入 nvcc –V,出现下面结果:

3.安装cuDNN:  cudnn-8.0-linux-x64-v5.1.solitairetheme8

百度cudnn下载进入,选择(需要自己注册账号):

进入下载目录

 cd /home/…

 sudo tar xvf cudnn-8.0-linux-x64-v5.1.solitairetheme8

 sudo cp cuda/include/cudnn.h /usr/local/cuda/include/  

 sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/  

 sudo chmod a+r /usr/local/cuda/include/cudnn.h  

 sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

4.安装BLAS

 sudo apt-get install libatlas-base-dev

5.安装opencv2.4.10(版本根据自己需求)

参考https://blog.csdn.net/oyster9/article/details/77917645

https://blog.csdn.net/Solomon1558/article/details/51967280

将步骤中的这个命令:

cmake -D CMAKE_BUILD_TYPE=RELEASE -D CMAKE_INSTALL_PREFIX=/usr/local ..

改为:

cmake -D CMAKE_BUILD_TYPE=release -D CMAKE_INSTALL_PREFIX=/usr/local -D CUDA_GENERATION=Kepler ..

安装前:

修改graphcuts.cpp文件(搜索即可,不修改会编译报错)

#if !defined (HAVE_CUDA) || defined (CUDA_DISABLER)

改成

#if !defined (HAVE_CUDA) || defined (CUDA_DISABLER) || (CUDART_VERSION >= 8000)

终端输入:

pkg-config --modversion opencv

出现对应版本号即安装成功。

6.安装caffe需要的python包

首先需要安装pip

sudo apt-get install python-pip

安装requirement里面的包

进入caffe的python目录,执行:

for req in $(cat requirements.txt); do pip install $req; done

打开.bashrc文件

 sudo gedit  ~/.bashrc

在文件末尾添加(添加caffe的python路径):

export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH(需要修改为自己的路径,如)

export PYTHONPATH=/home/myubuntu/download/caffe/python:$PYTHONPATH

保存。

运行source ~/.bashrc

7.编译caffe

在家目录执行

git clone https://github.com/BVLC/caffe.git

安装依赖项:

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

打开 caffe 所在目录,找到Makefile.config.example文件,将其改名为Makefile.config,打开。将# USE_CUDNN = 1一行开头的#删除,

在INCLUDE_DIRS :后面添加 /usr/local/cuda-8.0/include

在LIBRARY_DIRS :后面添加 /usr/local/cuda-8.0/lib

将CUDA_DIR := /usr/local/cuda-7.0改为cuda-8.0

保存。

打开终端:

make all

make test

make runtest

make pycaffe

验证:终端下输入

python

import caffe

不报错即成功。

参考教程:http://caffe.berkeleyvision.org/installation.html

          https://www.jianshu.com/p/69a10d0a24b9

          https://blog.csdn.net/LEILEI18A/article/details/79490081

          https://www.linuxidc.com/Linux/2015-07/120449.htm

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值