Loading...

Compile tensorflow on Jetson TX2

We use tensorflow in order to use deep learning algorithms such as Faster-RCNN, Yolo, VoxelChain.
Compiled .whl file is here.
If you want to install tensorflow into Jetson TX2, you should follow these instructions.
1. download jetson_tensorflow.zip
2. unzip jetson_tensorflow.zip
3. cd jetson_tensorflow
4. sh install.sh

We checked this .whl file by using this python code.(We get this code from here. Thank you!!)

import tensorflow as tf
# Creates a graph.
with tf.device('/gpu:2'):
  a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
  b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
  c = tf.matmul(a, b)
# Creates a session with allow_soft_placement and log_device_placement set
# to True.
sess = tf.Session(config=tf.ConfigProto(
      allow_soft_placement=True, log_device_placement=True))
# Runs the op.
print(sess.run(c))

If it works right, the output will be like this.

Leave a Reply