安装配置ubuntu14.04+cuda7.5+caffe

第一步:安装cuda,最好的办法就是在线安装:

打开cuda的官网安装:https://developer.nvidia.com/cuda-downloads

首先在官网上下载安装文件(链接前文已经提供):

cuda

我下载的是deb本地安装文件,下载完成后,按照文档提示的命令安装:

$ sudo dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb
$ sudo apt-get update
$ sudo apt-get install cuda

配置环境变量,我们直接放在系统配置文件profile里面,先打开profile文件

# sudo vi /etc/profile


  
  

在最后面加入两行代码,如果你还不会用vi进行编辑,请百度

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

保存退出,至此cuda 7.5安装完毕。



  
  

cuda安装完成之后可以简单的测试一下:

第一步:检查 CUDA Toolkit是否安装成功 
终端输入 :

nvcc -V
   
   
  • 1
  • 1

会输出CUDA的版本信息(V要大写) 
第二步:编译samples例子 
进入到Samples安装目录,然后在该目录下终端输入make,等待十来分钟。 
第三步:编译完成后测试 
可以在Samples里面找到bin/x86_64/linux/release/目录,并切换到该目录 
运行deviceQuery程序,sudo ./deviceQuery 
查看输出结果,重点关注最后一行,Pass表示通过测试 
这里写图片描述


第二步:安装caffe

   1.先下载caffe

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

然后安装一堆第三方库

复制代码
# sudo apt-get install libatlas-base-dev
# sudo apt-get install libprotobuf-dev
# sudo apt-get install libleveldb-dev
# sudo apt-get install libsnappy-dev
# sudo apt-get install libopencv-dev
# sudo apt-get install libboost-all-dev
# sudo apt-get install libhdf5-serial-dev
# sudo apt-get install libgflags-dev
# sudo apt-get install libgoogle-glog-dev
# sudo apt-get install liblmdb-dev
# sudo apt-get install protobuf-compiler
复制代码

2.接着,安装opencv

# cd caffe
# sudo git clone https://github.com/jayrambhia/Install-OpenCV
# cd Install-OpenCV/Ubuntu
# sudo sh dependencies.sh
# cd 2.4
# sudo sh opencv2_4_10.sh

如果过程中有报错,按照报错,去网上可以搜到相关的错误,进行解决!Python不用安装,ubuntu下面有2.7和3.5的版本

3.接下来,编译caffe

# cd ~/caffe
# sudo cp Makefile.config.example Makefile.config
# make all

至此,caffe安装完成。

4、配置运行环境

caffe运行时需要调用cuda的库,我们在/etc/ld.so.conf.d目录下新建一个cafe.conf文件,将所需要用的库的目录写入

# sudo vi /etc/ld.so.conf.d/caffe.conf

添加内容:

/usr/local/cuda/lib64

保存退出(:wq)

更新配置

# sudo ldconfig

5、测试caffe

  下载mnist数据

# cd ~/caffe
# sudo sh data/mnist/get_mnist.sh
# sudo sh examples/mnist/create_mnist.sh

运行时,如果你有GPU,则不需要修改配置文件,如果没有gpu,则需要修改配置文件lenet_solver.prototxt

# sudo vi examples/mnist/lenet_solver.prototxt

将最后一行的solver_mode:GPU改为solver_mode:CPU

配置好后,就可以运行了

# sudo sh examples/mnist/train_lenet.sh

注意,运行caffe程序时,必须在caffe的根目录下,不然会出错

注意:安装的过程中提示会缺少包,缺什么包就用

sudo apt-get install lib名;直接安装就可以了!

这样就可以玩起来ubuntu下的caffe,调用cuda加速了!


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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值