Tensorflow Lite от Nvidia Jetson

Question

Кто-нибудь использовал Tensorflow Lite на любом продукте Nvidia Jetson? Я хочу использовать мой Jetson Nano для вывода и хотел бы, чтобы с tf-lite использовался графический процессор.

Смущает, что , похоже, нет Python API для создания делегата с графическим процессором в tf-lite .

Есть ли для этого явные причины?

Является ли альтернативой использование полной библиотеки Tensorflow (я бы предпочел не использовать движок Nvidia TensorRT)?

Answer 1

Да, я пытался использовать tf lite на Jetson Nano раньше.

Вы можете сослаться на мою предыдущую статью на Medium (PS: я сожалею, что статья была написана на китайском языке.)

В этой статье рассказывается, как настроить среду TF Lite в Jetson Nano

Примечание:

Вам следует изменить следующую команду в соответствии со своей средой.

pip3 install https://dl.google.com/coral/python/tflite_runtime-2.1.0.post1-cp36-cp36m-linux_aarch64.whl

Настройка TF lite на Jetson Nano