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

Tensorflow 如何在像CUDA这样的桌面GPU上运行tflite模型?,tensorflow,gpgpu,Tensorflow,Gpgpu,我尝试在我的桌面ubuntu和Cuda GPU上运行mobilenet_v2_1.0_224.tflite模型。 但默认情况下,它只在CPU上运行 我怎样才能让它也在cuda GPU上运行 顺便说一句,我能够使用nnapi在移动设备加速器上运行相同的功能。 问候,

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

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

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