TX2源码安装编译mxnet踩坑记录

Install mxnet for cpp package in TX2 is not that easy

Record here for my experience.

Steps:

Follow the documentation on this site Install MXNet on a Jetson , there is a little different

1. First clone mxnet from github && cd mxnet

   git clone --recursive https://github.com/apache/incubator-mxnet.git mxnet
   cd mxnet
   git submodule init
   git submodule update
   

2. Configure CUDA:

   nvcc --version
   

   on my TX2 is CUDA9.0

   sudo rm /usr/local/cuda
   sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda
   

   cudnn v7.1.3

3. Copy config.mk

     cp make/crosscompile.jetson.mk config.mk
   

4. Edit config.mk, in config.mk , modify these settings:

   1. USE_CUDA_PATH = /usr/local/cuda

   2. USE_OPENCV = 1

   3. USE_JEMALLOC = 0 which is different from official guide but VERY IMPORTENT

   4. USE_GPERFTOOLS = 0 which is different from official guide but **VERY IMPORTENT

   5. USE_CPP_PACKAGE = 1 for cpp package

   6. Update the NVCC settings. NVCCFLAGS := -m64

   there 3 and 4 is important , or when