标签: tensorflow gpgpu
我尝试在台式机ubuntu和Cuda GPU上运行mobilenet_v2_1.0_224.tflite模型。 但是默认情况下,它仅在CPU上运行。
如何使它也可以在cuda GPU上运行?
BTW,能够使用nnapi在移动设备加速器上运行相同的命令。 问候