Build Tensorflow1.5.0 From Jetson TX

6241 ワード

Author: Mikoy Date: 2018-03-16

1. Preparation:

  • Jetpack 3.2
  • Download file from JetPack3.2.

  • Java
  • Bazel

  • 2. Installing Step:


    Step 1: Install JetPack3.2

  • Download this file and install like this Url.

  • Step 2: Install Java

    sudo add-apt-repository ppa:webupd8team/java
    sudo apt-get update
    sudo apt-get install oracle-java8-installer
    

    Step 3: Install More Stuff

    # For Python2
    sudo apt-get install zip unzip autoconf automake libtool curl zlib1g-dev maven -y
    sudo apt-get install python-numpy swig python-dev python-pip python-wheel -y
    
    # For Python3
    sudo apt-get install zip unzip autoconf automake libtool curl zlib1g-dev maven -y
    sudo apt-get install python3-numpy swig python3-dev python3-pip python3-wheel -y
    

    Step 4: Install Bazel

  • Download the Bazel-0.11.1-dist
  • Unzip the package
  • # Up to your download file path.
    sudo unzip ./bazel.0.11.1-dist.zip -d ./bazel.0.11.1-dist
    
  • Cd bazel-0.11.1-dist
  • cd ./bazel.0.11.1-dist
    
  • Start the compilation process by issuing
  • ./compile.sh
    

      If you get the error : Closure Rules requires Bazel >=0.4.5 but was 0.11.1 , see like this bazel#4834. Change all the error file the version 0.4.5 to 0.0.0. Then you will build it successfully.   If you get the error : /home/nvidia/.cache/bazel/_bazel_nvidia/d2751a49dacf4cb14a513ec663770624/external/io_bazel_rules_closure/closure/stylesheets/closure_css_library.bzl:27:13: The function 'set' has been removed in favor of 'depset', please use the latter. You can temporarily refer to the old 'set' constructor from unexecuted code by using --incompatible_disallow_uncalled_set_constructor=false , see like this issue bazel#4828. Build it again in another time.
  • Copy the build to your system bin folder
  • sudo cp output/bazel /usr/local/bin
    

    Step 5: Configure Tensorflow

  • Download Tensorflow1.5.0
  • #Download From https://github.com/tensorflow/tensorflow.git
    git clone https://github.com/tensorflow/tensorflow.git
    cd ./tensorflow
    git checkout v1.5.0
    # Change some INFO for Jetson TX
    sudo gedit tensorflow/stream_executor/cuda/cuda_gpu_executor.cc
    #In the function static intTryToReadNumaNode(conststring &pci_bus_id,intdevice_ordinal)add the following lines at the start of the function. 
    #ADD just like this(you add it into you edit file where the function is.):
        LOG(INFO) << "ARM has no NUMA node, hardcoding to return zero";
        return 0;
    #Then cp the cudnn.h
    sudo mkdir /usr/lib/aarch64-linux-gnu/include/
    sudo cp /usr/include/cudnn.h /usr/lib/aarch64-linux-gnu/include/cudnn.h
    
  • Configure Tensorflow
  • ./configure
    #Then configure it what you need. There are my options.
    Please specify the location of python. [Default is /usr/bin/python]: /usr/bin/python3
    Please specify optimization flags to use during compilation [Default is -march=native]: 
    Do you wish to use jemalloc as the malloc implementation? (Linux only) [Y/n] y
    jemalloc enabled on Linux
    Do you wish to build TensorFlow with Google Cloud Platform support? [y/N] n
    No Google Cloud Platform support will be enabled for TensorFlow
    Do you wish to build TensorFlow with Hadoop File System support? [y/N] n
    No Hadoop File System support will be enabled for TensorFlow
    Do you wish to build TensorFlow with the XLA just-in-time compiler (experimental)? [y/N] n
    XLA JIT support will be enabled for TensorFlow
    Found possible Python library paths:
      /usr/local/lib/python2.7/dist-packages
      /usr/lib/python2.7/dist-packages
    Please input the desired Python library path to use.  Default is [/usr/local/lib/python3/dist-packages]
    
    Using python library path: /usr/local/lib/python2.7/dist-packages
    Do you wish to build TensorFlow with OpenCL support? [y/N] n
    No OpenCL support will be enabled for TensorFlow
    Do you wish to build TensorFlow with CUDA support? [y/N] y
    CUDA support will be enabled for TensorFlow
    Please specify which gcc should be used by nvcc as the host compiler. [Default is /usr/bin/gcc]: 
    Please specify the CUDA SDK version you want to use, e.g. 7.0. [Leave empty to use system default]: 7.0.5
    Please specify the location where CUDA  toolkit is installed. Refer to README.md for more details. [Default is /usr/local/cuda]: 
    Please specify the Cudnn version you want to use. [Leave empty to use system default]: 
    Please specify the location where cuDNN  library is installed. Refer to README.md for more details. [Default is /usr/local/cuda]: /usr/lib/aarch64-linux-gnu
    Please specify a list of comma-separated Cuda compute capabilities you want to build with.
    You can find the compute capability of your device at: https://developer.nvidia.com/cuda-gpus.
    Please note that each additional compute capability significantly increases your build time and binary size.
    Extracting Bazel installation...
    .......................
    INFO: Starting clean (this may take a while). Consider using --expunge_async if the clean takes more than several minutes.
    .......................
    INFO: All external dependencies fetched successfully.
    Configuration finished
    
  • Once your configuration is done, run like this:
  • bazel build -c opt --local_resources 3072,4.0,1.0 --verbose_failures --config=cuda //tensorflow/tools/pip_package:build_pip_package
    
  • Once Tensorflow is compiled, build the pip package:
  • bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
    
  • Move the pip wheel from the tmp directory if you want to save it:
  • mv /tmp/tensorflow_pkg/tensorflow-1.5.0-cp35-cp35mu-linux_aarch64.whl $HOME
    
  • Install the pip wheel:
  • sudo pip3 install $HOME/tensorflow-1.5.0-cp35-cp35mu-linux_aarch64.whl
    

    3. Test the Tensorflow

    python3
    import tensorflow as tf
    tf.__version__
    >>1.5.0