安裝TensorFlow

Search…
安裝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!
安裝OpenCV
廣義線性模型 Generalized Linear Models
Copy link