caffe安装

来源:互联网 发布:windows firewall 编辑:程序博客网 时间:2024/04/28 17:13

通用的依赖库

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compilersudo apt-get install --no-install-recommends libboost-all-dev

cuda

1.在NVIDIA官网下载最新版本cuda_7.5.18_linux.run

chmod +x ./cuda_7.5.18_linux.run./cuda_7.5.18_linux.run --extract=~/Desktop

得到三个单独的安装包:
NVIDIA显卡驱动:NVIDIA-Linux-x86_64-352.39.run
cuda安装包:cuda-linux64-rel-7.5.18-19867135.run
cuda例程安装包:cuda-samples-linux-7.5.18-19867135.run
如果集成的显卡驱动版本过低,可在官网单独下载。
2.禁用自带的nouveau显卡驱动

sudo gedit /etc/modprobe.d/blacklist.conf#在文件中添加blacklist nouveau#保存并退出,重启

3.安装NVIDIA显卡驱动
ctrl+alt+F1进入tty,登录后进入安装包所在目录

sudo stop lightdmchmod +x ./NVIDIA-Linux-x86_64-352.39.runsudo ./NVIDIA-Linux-x86_64-352.39.runsudo start lightdm

4.安装cuda
进入安装包所在路径

chmod +x ./cuda-linux64-rel-7.5.18-19867135.runsudo ./cuda-linux64-rel-7.5.18-19867135.run

5.添加环境变量

#打开配置文件sudo gedit /etc/profile#在文件最后添加#PATH=/usr/local/cuda-6.5/bin:$PATH#export PATH#保存并退出,执行下列命令,使更改立即生效source /etc/profile

6.添加lib库路径

#创建配置文件sudo gedit /etc/ld.so.conf.d/cuda.conf#在文件中输入#/usr/local/cuda-7.5/lib64#保存并退出,执行下列命令,使更改立即生效sudo ldconfig

opencv

ffmpeg移出了ubuntu14.04的官方库,因此无法用apt-get到ffmpeg及相关库,需要手动添加PPA,ubuntu15.04无需进行这步操作。

sudo add-apt-repository ppa:mc3man/trusty-mediasudo apt-get update

Github上有人已经写好了opencv完整的安装脚本

cd ~/caffegit clone https://github.com/jayrambhia/Install-OpenCVcd Install-OpenCV/Ubuntu/2.4sudo sh ./opencv2_4_10.sh

opencv249编译中可能会有以下错误

Generating precomp.hpp[ 15%] nvcc fatal   : Unsupported gpu architecture 'compute_11'CMake Error at cuda_compile_generated_matrix_operations.cu.o.cmake:208 (message):  Error generating  /home/hiki/caffe/Install-OpenCV/Ubuntu/2.4/OpenCV/opencv-2.4.9/build/modules/core/CMakeFiles/cuda_compile.dir/__/dynamicuda/src/cuda/./cuda_compile_generated_matrix_operations.cu.o

解决方法
- 在opencv2_4_9.sh中的cmake添加”-D CUDA_GENERATION=Kepler”;
- 或者按照opencv2_4_10.sh的cmake设置显卡架构,暂未测试可行性。


BLAS

caffe默认配置的BLAS是ATLAS,也可以安装OpenBLAS或者MKL获得更好的性能。

 sudo apt-get install libatlas-base-dev

matlab

  1. 将iso文件挂载到ubuntu上:右键iso文件 -> Open With -> Disk Image Mounter;
  2. 进入挂载目录,如:cd /media/hiki/MATHWORKS_R2014B/
  3. 安装:sudo ./install
  4. 安装时选择离线安装,序列号见crack目录下readme.txt;
  5. 激活时选择离线激活,激活文件见crack目录下license.lic;
  6. 将crack目录下libmwservices.so复制到/usr/local/MATLAB/R2014b/bin/glnxa64/下,替换已有文件:sudo cp libmwservices.so /usr/local/MATLAB/R2014b/bin/glnxa64/
  7. 安装matlab support:sudo apt-get install matlab-support
  8. 运行matlab:sudo matlab

caffe

1.安装依赖库

sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

2.编译

cd ~/caffeunzip caffe-master.zipcd caffe-mastercp Makefile.config.example Makefile.configmake all -j4#注意 -j4 是指使用几个线程来同时编译

3.测试

cd ~/caffe/caffe-mastersh data/mnist/get_mnist.shsh examples/mnist/create_mnist.shtime sh examples/mnist/train_lenet.sh

可以在 ~/caffe/caffe-master/examples/mnist/lenet_solver.prototxt 文件中运行 CPU 版本或者 GPU 版本。修改最后一行, solver_mode: CPU 或者 solver_mode: GPU

0 0
原创粉丝点击