您的位置:首页 > Web前端

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

2017-06-10 11:03 375 查看

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

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

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



我下载的是
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[/code] 
然后安装一堆第三方库

# 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加速了!
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: