NVIDIA Jetson TX1刷机教程与caffe配置(有点潦草,个人觉得有帮助)

    xiaoxiao2021-03-25  105

    刷机教程参照:http://blog.csdn.net/Jalong_Ma/article/details/52743923里面的介绍即可。

    caffe配置:

    caffe的配置可能遇到很多问题,我推荐的步骤是:

    参照http://blog.csdn.net/Jalong_Ma/article/details/52743923安装依赖库,然后:

    安装NCCL:

    $ git clone https://github.com/NVIDIA/nccl.git $ cd nccl

    $ sudo make install -j4

    NCCL 库和文件头将安装在 /usr/local/lib 和 /usr/local/include 中。

    编译caffe: git clone https://github.com/BVLC/caffe.git cd caffe cp Makefile.config.example Makefile.config 修改Makefile.config: # cuDNN acceleration switch (uncomment to build with cuDNN). USE_CUDNN := 1 # Uncomment if you're using OpenCV 3 OPENCV_VERSION := 3 # Uncomment to support layers written in Python (will link against Python libs) WITH_PYTHON_LAYER := 1  此外还需要: 把#USE_NCCL := 1改为: USE_NCCL := 1  TX1的GPU计算能力是53,所以更改Makefile.config文件中更改:CUDA_ARCH 下加入一行:-gencode arch=compute_53,code=sm53 \ 由于hdf5库目录更改,所以需要单独添加: 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/ (单独添加的意思就是把上面这两行复制到MaKefile.config文件里即可)。 之后便可以编译: make -j4 all make -j4 runtest .......按照上面的参考博客做就行。

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

    最新回复(0)