How to install tensorflow cpu version on Jetson TX2

Robotics & Edge Computing Jetson & Embedded Systems Jetson TX2
1

I have tested some methods to install TensorFlow CPU version on TX2. For example, by pip install TensorFlow-aarch64, but after this, I can not import TensorFlow. I also downloaded some pre-compiled aarch64 Tensorflow wheel (some in https://github.com/lhelontra/tensorflow-on-arm/releases), but it can not be installed on TX2, as it doesn’t support the TX2 platform. Please help me, thank you.

2

Hi,

You can just install our TensorFlow package:
https://docs.nvidia.com/deeplearning/frameworks/install-tf-jetson-platform/index.html#prereqs

Then turn off the GPU mode.

import os
os.environ["CUDA_VISIBLE_DEVICES"]="-1"

Thanks.

3

Hi,

Thank you. I will try.