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.