Я пытаюсь запустить модель mobilenet_v2_1.0_224.tflite на настольном компьютере с Ubuntu и графическим процессором Cuda.
Но по умолчанию он работает только на процессоре.
Как мне сделать так, чтобы он работал и на cuda GPU?
Кстати, я могу запустить то же самое на ускорителях мобильных устройств, используя nnapi.
С уважением,