如何在像 CUDA 这样的桌面 GPU 上运行 tflite 模型?

Alb*_*Joy 5 gpgpu tensorflow

我尝试在桌面 ubuntu 和 Cuda GPU 上运行 mobilenet_v2_1.0_224.tflite 模型。但默认情况下它仅在 CPU 上运行。

我怎样才能让它在 cuda GPU 上运行?

顺便说一句,我能够使用 nnapi 在移动设备加速器上运行相同的内容。问候,