JetsonHacks

Install TensorFlow for Python – NVIDIA Jetson TX Dev Kits

Install TensorFlow

Install TensorFlow

The last few articles we’ve been building TensorFlow packages which support Python. The packages are now in a Github repository, so we can install TensorFlow without having to build it from source. Install files are available both for the Jetson TX1 and Jetson TX2. Looky here:

Background

In the earlier articles we went over the procedure to building TensorFlow for the Jetson TX1 and the Jetson TX2. The procedure takes a few hours on each platform. Worse, for the Jetson TX1 a new kernel is needed so that swap memory can be enabled.

Fortunately we were able to build a new Github repository called installTensorFlowJetsonTX which contains all of the Python install .whl (wheel) files that were created during the process (and then some).

The repository contains two sets of wheel files. There is a set of files for the Jetson TX1 in the TX1 directory. The other set of wheel files are for the Jetson TX2, which is in the TX2 directory. Each of the directories contains two wheel files. One of the wheel files is for Python 2.7, the other is for Python 3.5. For example, the file tensorflow-1.3.0-cp35-cp35m-linux_aarch64.whl in the TX1 folder is the TensorFlow 1.3 wheel file for Python 3.5. The term aarch64 indicates that the file is for the Jetson TX1 ARM64 architecture.

Install TensorFlow

Install the wheel files with then Python python-pip tool. The first step is to install the appropriate version python-pip. Next, download the appropriate wheel file from the repository. Then use the pip tool with the corresponding wheel file to finish the installation.

For Python 2.7

$ sudo apt-get install -y python-pip python-dev
$ pip install tensorflow-wheel-file

Python 3.5

$ sudo apt-get install -y python3-pip python3-dev
$ pip3 install tensorflow-wheel-file

Downloading the wheel file can be done in a couple of ways. The first way is to clone the repository. This has the side effect of downloading all of the wheel files (currently about 200MB). Perhaps a better way, as shown in the video, is to navigate to the wheel file you wish to download in the web browser on the Github site, and then click the Download button. The wheel file will be placed in your Downloads folder.

Conclusion

This is a relatively easy way to install TensorFlow. TensorFlow is a very rich environment, the downside of this method is that you will not be able to specify the build options. Note that you will also have to install the TensorFlow models and support materials. The upside is that you don’t have to spend the build time just to get TensorFlow up and running.

Notes

There are wheel files for Python 2.7 and Python 3.5 The Jetson environment for both the TX1 and TX2:

TensorFlow

Repositories used for building the wheel files:

NVIDIA Jetson TX1: https://github.com/jetsonhacks/installTensorFlowTX1
NVIDIA Jetson TX2: https://github.com/jetsonhacks/installTensorFlowTX2

Note: The Jetson TX1 GPU architecture is 5.3, the Jetson TX2 is 6.2. The builds reflect this.

Skip to toolbar