Use to use GPU if available on mobile

Hi

I have neural network that I trained on my PC and then jit compiled and run inferences on my Android phone. I can see from Android Studio debugger it is processing using the mobile’s CPU. How can I check and use GPU if available on android mobile?

Thanks in advance

Hi,

Unfortunately, mobile GPU is a completely different language and architecture to everything else. So we don’t support it yet.
It is work in progress though and you can see all the PRs working it here: https://github.com/pytorch/pytorch/pulls?q=is%3Apr+is%3Aopen+vulkan

1 Like