博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
ubuntu14.04配置caffe
阅读量:6719 次
发布时间:2019-06-25

本文共 2034 字,大约阅读时间需要 6 分钟。

ubuntu 14.04 64bit 安装

请自行Google安装,并修改源。

cuda 7.5 安装

  1. ,选择ubuntu 14.04的deb包下载

  2. 安装一些可能的依赖

    sudo apt-get install linux-headers-$(uname -r) build-essential

  3. 安装cuda

    sudo dpkg –i cuda-repo-<distro>_<version>_<architecture>.deb

    sudo apt-get update

    sudo apt-get install cuda

    注: cuda的安装,并不需要先手动安装NVIDIA驱动,安装cuda会依赖NVIDIA驱动,请保持联网,会自行下载安装。

  4. 环境变量设置

    在/etc/profile文件加入

    PATH=/usr/local/cuda-7.5/bin:$PATH

    export PATH

    然后输入source /etc/profile使更改生效

  5. 安装cuda_samples

    cuda-install-samples-7.5.sh <dir>

    cd ~/NVIDIA_CUDA-7.5_Samples

    make -j8 编译

    切换到bin目录,执行./deviceQuery,显示Result=PASS表示安装成功

  6. cudnn加速安装

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

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

    cd /usr/local/cuda/lib64

    sudo rm -rf libcudnn.so.7.0 libcudnn.so

    sudo ln -s libcudnn.so.7.0.64 libcudnn.so.7.0

    sudo ln -s libcudnn.so.7.0 libcudnn.so

    更改文件权限

  7. 添加lib路径

    sudo vim /etc/ld.so.conf.d/cuda.conf

    输入 /usr/local/cuda/lib64

    保存,并执行sudo ldconfig

caffe 编译

  1. 安装依赖

    sudo apt-get install -y libblas-dev liblapack-dev libopenblas-base libopenblas-dev

    sudo apt-get install -y libboost1.55-all-dev

    sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev

    sudo apt-get install -y libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler

  2. 修改Makefile.config

    cp Makefile.config.example Makefile.config

    修改如下

    1. 取消# USE_CUDNN := 1注释
    2. 修改PYTHON_INCLUDE路径
    3. 修改MATLAB_DIR路径
    4. 取消# WITH_PYTHON_LAYER := 1注释
    5. 修改BLAS := open
  3. 编译并测试

    make all -j8

    make test -j8

    make runtest -j8

Python caffe 接口编译

  1. 安装python模块

    sudo apt-get install python-dev python-pip gfortran

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

    如果有模块安装不成功可以多执行几次。

    pip可以修改源,加快下载速度,国内有豆瓣、v2ex等python源。使用方法,自行google。

  2. 编译pycaffe

    make pycaffe -j8

  3. ~/.bashrc中添加PYTHONPATH变量

    export PYTHONPATH=$PYTHONPATH:/home/whq/deep-learning/caffe-master/python

  4. 执行make pytest -j8测试

Matlab caffe 接口编译

  1. 安装MATLAB
  2. (MATLAB 2014a,只支持gcc4.7)或
  3. 编译matcaffe

    make matcaffe -j8

  4. 执行make mattest -j8测试


机器硬件配置为i7 4790x、 32GB RAM、GeForce GTX TITAN,因软件硬件环境不同可能造成安装失败。

如有问题,欢迎留言指正。

转载于:https://www.cnblogs.com/wuhanqing/p/5359073.html

你可能感兴趣的文章
HP Instant Information
查看>>
Maven 手动添加 JAR 包到本地仓库
查看>>
电子辅助的个体化严密控制策略比常规方法更有效地帮助早期RA实现全面控制病情[EULAR2015_THU0122]...
查看>>
【BZOJ3529】数表
查看>>
最坏的不是面试被拒,而是没面试机会,以面试官视角分析哪些简历至少能有面试机会...
查看>>
python的文件操作与目录操作os模块
查看>>
客户端存储cookie
查看>>
agal常用命令
查看>>
html转义
查看>>
自定义UITextView实现placeholder效果
查看>>
格式化
查看>>
如何让字体大小<12px
查看>>
【机器学习】--主成分分析PCA降维从初识到应用
查看>>
对于数组的操作:splice与slice
查看>>
第二次Scrum meeting
查看>>
云安全
查看>>
addEventListener调用带参数函数
查看>>
postgresql----INSERT
查看>>
MVC中渲染页面
查看>>
Java OA的优势何在
查看>>