博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
Caffe2(1)----Ubantu14.04安装
阅读量:4657 次
发布时间:2019-06-09

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

英文好的请直接参考官方安装文档:

Caffe2的安装相比以前Caffe一代的安装,简直有点一键装机的感觉,下面简单总结下Caffe2的安装。

环境:Ubantu14.04

安装步骤:

1.依赖库安装

sudo apt-get updatesudo apt-get install -y --no-install-recommends \      build-essential \      cmake \      git \      libgoogle-glog-dev \      libprotobuf-dev \      protobuf-compiler \      python-dev \      python-pip                          sudo pip install numpy protobuf

2.GPU库安装(可选)

a.CUDA安装

sudo apt-get update && sudo apt-get install wget -y --no-install-recommendswget "http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_8.0.61-1_amd64.deb"sudo dpkg -i cuda-repo-ubuntu1404_8.0.61-1_amd64.debsudo apt-get updatesudo apt-get install cuda

b.cuDNN 5.1安装

CUDNN_URL="http://developer.download.nvidia.com/compute/redist/cudnn/v5.1/cudnn-8.0-linux-x64-v5.1.tgz"wget ${CUDNN_URL}sudo tar -xzf cudnn-8.0-linux-x64-v5.1.tgz -C /usr/localrm cudnn-8.0-linux-x64-v5.1.tgz && sudo ldconfig

3.其它依赖(可选)

sudo apt-get install -y --no-install-recommends libgflags2
sudo apt-get install -y --no-install-recommends \      libgtest-dev \      libiomp-dev \      libleveldb-dev \      liblmdb-dev \      libopencv-dev \      libopenmpi-dev \      libsnappy-dev \      openmpi-bin \      openmpi-doc \      python-pydotsudo pip install \      flask \      graphviz \      hypothesis \      jupyter \      matplotlib \      pydot python-nvd3 \      pyyaml \      requests \      scikit-image \      scipy \      setuptools \      tornado

4.下载和编译

git clone --recursive https://github.com/caffe2/caffe2.git && cd caffe2make && cd build && sudo make install

5.测试是否安装成功

python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"

如果成功,显示Success,恭喜安装成功;否则显示Failure。如果失败则可能是Python环境变量甚至问题,具体可在文件~/.bashrc(可使用 gedit或vim打开)中添加Python的环境变量代码,然后重启下终端,如下所示:

export PYTHONPATH=/usr/local:$PYTHONPATHexport PYTHONPATH=$PYTHONPATH:/home/ubuntu/caffe2/buildexport LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH

整个过程一般没什么问题,看来Caffe2确实已经比较接近产品级开发的使用了。 

转载于:https://www.cnblogs.com/cv-pr/p/6925124.html

你可能感兴趣的文章
HDU 3374 String Problem
查看>>
数据集
查看>>
[Leetcode] unique paths ii 独特路径
查看>>
HDU 1217 Arbitrage (Floyd + SPFA判环)
查看>>
IntelliJ idea学习资源
查看>>
Django Rest Framework -解析器
查看>>
ExtJs 分组表格控件----监听
查看>>
Hibernate二级缓存配置
查看>>
LoadRunner常用术语
查看>>
关于jedis2.4以上版本的连接池配置,及工具类
查看>>
记忆讲师石伟华微信公众号2017所有文章汇总(待更新)
查看>>
FactoryBean
查看>>
Coolite动态加载CheckboxGroup,无法在后台中获取
查看>>
C3P0连接池工具类使用
查看>>
SVN常用命令备注
查看>>
孩子教育
查看>>
解决Cacti监控图像断断续续问题
查看>>
结构体的传参理解成员的存储方式
查看>>
python 进程与线程(理论部分)
查看>>
什么是API
查看>>