安裝TensorFlow
    1.
    安裝依賴套件
    1
    $ sudo apt-get install default-jdk libcupti-dev
    2
    $ export JAVA_HOME='/usr/lib/jvm/java-8-openjdk-arm64/'
    Copied!
    2.
    取得 TensorFlow 編譯腳本
    1
    $ git clone git://github.com/jetsonhacks/installTensorFlowTX2
    2
    $ cd installTensorFlowTX2
    Copied!
    3.
    執行編譯腳本
    1
    $ ./installPrerequisitesPy3.sh
    2
    $ ./cloneTensorFlow.sh
    3
    $ ./setTensorFlowEVPy3.sh
    4
    $ ./buildTensorFlow.sh
    5
    $ ./packageTensorFlow.sh
    Copied!
    4.
    安裝 TensorFlow
      1.
      建立虛擬開發環境
      1
      $ virtualenv TensorFlow
      Copied!
      2.
      進入虛擬開發環境
      1
      $ cd TensorFlow
      2
      $ source bin/active
      Copied!
      3.
      安裝 TensorFlow 到虛擬環境
      1
      pip3 install $HOME/<TensorFlow 的 .whl 安裝封包>
      Copied!
      4.
      測試 TensorFlow
        1.
        Hello World
        1
        import tensorflow as tf
        2
        hello = tf.constant('Hello, TensorFlow on NVIDIA Jetson TX2!')
        3
        sess = tf.Session()
        4
        print(sess.run(hello))
        Copied!
        輸出
        1
        Hello, TensorFlow on NVIDIA Jetson TX2!
        Copied!
        2.
        運算單元
        1
        import tensorflow as tf
        2
        # Creates a graph.
        3
        a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
        4
        b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
        5
        c = tf.matmul(a, b)
        6
        # Creates a session with log_device_placement set to True.
        7
        sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
        8
        # Runs the op.
        9
        print(sess.run(c))
        Copied!
        輸出
        1
        name: NVIDIA Tegra X2
        2
        major: 6 minor: 2 memoryClockRate (GHz) 1.3005
        3
        MatMul: (MatMul): /job:localhost/replica:0/task:0/gpu:0
        4
        b: (Const): /job:localhost/replica:0/task:0/gpu:0
        5
        a: (Const): /job:localhost/replica:0/task:0/gpu:0
        6
        [[ 22. 28.]
        7
        [ 49. 64.]]
        Copied!
Copy link