Поддержка графического процессора Tensorflow Lite для Python

Question

Кто-нибудь знает, есть ли в Tensorflow Lite поддержка графического процессора для Python?Я видел руководства для Android и iOS, но ничего о Python я не встречал.Если установлена ​​tensorflow-gpu и импортирована tensorflow.lite.python.interpreter, будет ли автоматически использоваться графический процессор?

Answer 1

Вы можете заставить вычисления выполняться на GPU:

import tensorflow as tf
with tf.device('/gpu:0'):
   for i in range(10):
         t = np.random.randint(len(x_test) )
         ...

Надеюсь, это поможет.