TFLite Model is not using GPU

Autonomous Machines Jetson & Embedded Systems Jetson Nano
1

Hi, i’ve installed TensorFlow v2.5.0 on my Jetson Nano following these instructions Installing TensorFlow for Jetson Platform - NVIDIA Docs . I use tegrastats command to monitor GPU usage: from that i can see that TF model is running on GPU but the TFLite model keeps GPU to 0%. I load TFLite interpreter using

interpeter = tf.lite.Interpreter(model_path=‘lite_model.tflite’)

Did I skip some step? Is there a way to fix it? Or doesn’t Jetson Nano have GPU Support for TensorFlow Lite?

Thanks

3

Hi,

Please check the document below.
TFLite GPU delegate only supports Android and iOS GPU.

Thanks.

5

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.