Hi, my goal is to deploy CNNs (or Conv RNNs) on Android devices with Snapdragon processors. ex) Pixel 4 (snapdragon 855)
I want to make my models run as fast as possible with GPU or DSP.
Currently, I find 2 ways to deploy PyTorch models.
- PyTorch mobile
This is a natively supported method of PyTorch and I like it.
But mobile GPU execution is not supported yet.
I see several PRs related to vulkan APIs here
Will mobile GPU/DSP be supported in near future?
I found this article from Qualcomm that they will support PyTorch but I don’t find any follow-up.
- PyTorch -> onnx
Pytorch models could be exported to onnx format.
From onnx I can convert the models into DLC format with SNPE SDK.
But it will be more difficult to create an app with such forms.
Are there any other ways I didn’t find? What would be the best way to deploy PyTorch models on mobile devices?
I used TensorFlow lite before but would prefer to use PyTorch if possible as I mainly use PyTorch.