How to detect the image on GPU?

[KingWangJL]

In my project,i'm using a edge device.This is a dual system device,it includes Android and linux system.i want this demo in the derive.i has gpu.so i wantto know How to run on GPU.Thank you very much。