Skip to content Skip to sidebar Skip to footer
Home / Interpreter / Python / Tensorflow / Tensorflow Lite

How Can I Use Gpu For Running A Tflite Model (*.tflite) Using Tf.lite.interpreter (in Python)?

Post a Comment
I have converted a tensorflow inference graph to tflite model file (*.tflite), according to instructions from https://www.tensorflow.org/lite/convert. I tested the tflite model on

Solution 1:

https://github.com/tensorflow/tensorflow/issues/34536

CPU is kind of good enough for tflite, especially multicore.

nvidia GPU likely not updated for tflite, which is for mobile GPU platform.

Solution 2:

Conspiracy: they (TF-NVIDIA) hand-shake to not let TFlite working on GPU ? oo easy to make one.

Steve

Post a Comment for "How Can I Use Gpu For Running A Tflite Model (*.tflite) Using Tf.lite.interpreter (in Python)?"