基于Jetson TX1的caffe环境搭建

    xiaoxiao2021-03-25  63

    Data: 2017.03.13 Author: cjh

    Theme: Set up the Caffe environment

    步骤1:安装caffe所需的安装包

     $ sudo add-apt-repository universe  $ sudo add-apt-repository multiverse  $ sudo apt-get update  $ sudo apt-get install libboost-all-dev libprotobuf-dev libleveldb-dev libsnappy-dev  $ sudo apt-get install libhdf5-serial-dev protobuf-compiler libgflags-dev libgoogle-glog-dev  $ sudo apt-get install liblmdb-dev libblas-dev libatlas-base-dev

    步骤2:下载caffe

    1. git clone git://github.com/BVLC/caffe.git

    2.若网速不够可在windows上下载

        https://github.com/BVLC/caffe

        然后复制到你的开发板上如下命令

        $ mkdir -pv $HOME/Work/caffe     $ cp caffe-master.zip $HOME/Work/caffe/     $ cd $HOME/Work/caffe/ && unzip caffe-master.zip

        

    步骤3:编译caffe

         $ cd $HOME/Work/caffe/caffe-master      $ cp Makefile.config.example Makefile.config

         $ vim Makefile.config

          修改Makefile.config文件

          USE_CUDNN := 1

          INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial       LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/aarch64-linux-gnu/hdf5/serial

    有必要课修改cuda的路径,我的开发板用的是cuda-8.0

    所以我cuda的路径修改为了/usr/local/cuda-8.0

    修改完成后保存并退出

    $ make -j4

    经过以上步骤应该就能完成caffe的环境搭建了

    转载请注明原文地址: https://ju.6miu.com/read-40429.html

    最新回复(0)