如何在台式机GPU(如CUDA)上运行tflite模型?

时间:2019-03-15 05:00:50

标签: tensorflow gpgpu

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

如何使它也可以在cuda GPU上运行?

BTW,能够使用nnapi在移动设备加速器上运行相同的命令。 问候

0 个答案:

没有答案