Official TensorFlow for Jetson AGX XavierNX

Here is the information of official TensorFlow release for XavierNX.

Python 3.6+JetPack4.4

$ sudo apt-get update
$ sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev liblapack-dev libblas-dev gfortran
$ sudo apt-get install python3-pip
$ sudo pip3 install -U pip testresources setuptools
$ sudo pip3 install -U numpy==1.16.1 future==0.17.1 mock==3.0.5 h5py==2.9.0 keras_preprocessing==1.0.5 keras_applications==1.0.8 gast==0.2.2 futures protobuf pybind11
# TF-2.x
$ sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v44 tensorflow==2.2.0+nv20.7
# TF-1.15
$ sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v44 ‘tensorflow<2’

Thanks.

1 Like

TensorFlow Installation Documentation - https://docs.nvidia.com/deeplearning/frameworks/install-tf-jetson-platform/index.html

Does JetPack 4.4 support for TF C API?

Hi,
I meet the error , showed as below.


So, which one should I select?

So, how to get the version of nv ?

Hi,

Please try this command:

$ sudo pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v44 tensorflow==1.15.2+nv20.6

Thanks.

Hi,

Sorry that we don’t support C++ API in our latest package.
This is in our feature request list and will be prioritized by our internal team.

Thanks.