Hi Rahul,
I am experiencing the same problem. I started this project to document down all the steps, issues and problems I experience along the way in my attempt to ship a neural network developed in PyTorch 1.0 nightly to Android devices:

  
      github.com
  
  
    cedrickchee/data-science-notebooks/blob/master/notebooks/deep_learning/fastai_mobile/README.md 137
# Fast.ai Mobile Camera

:tada: Check out a working PyTorch-Caffe2 implementation on mobile: :tada:

- [Android Camera app demo (video)](https://youtu.be/TYkoaVNCMos)

**Guide - How I Shipped a Neural Network on Android/iOS Phones with PyTorch and Android Studio/Xcode**

I'll walk you through every step, from problem all the way to building and deploying the Android/iOS app to mobile phones.

- Learn how to ship SqueezeNet from PyTorch to Caffe2 + Android/iOS app. Please take a look at this [notebook](https://nbviewer.jupyter.org/github/cedrickchee/data-science-notebooks/blob/master/notebooks/deep_learning/fastai_mobile/shipping_squeezenet_from_pytorch_to_android.ipynb).
  - Android project for AI Camera app tutorial in this [notebook](https://nbviewer.jupyter.org/github/cedrickchee/data-science-notebooks/blob/master/notebooks/deep_learning/fastai_mobile/shipping_squeezenet_from_pytorch_to_android.ipynb#Fast.ai-Mobile-Camera-Project).
  - iOS project (TBD)
- Get started with an introduction to [Open Neural Network Exchange format (ONNX)](https://onnx.ai/) in this Jupyter [notebook](https://nbviewer.jupyter.org/github/cedrickchee/data-science-notebooks/blob/master/notebooks/deep_learning/fastai_mobile/onnx_from_pytorch_to_caffe2.ipynb).

[Source code for the Android app](https://github.com/cedrickchee/pytorch-android).

---

## Background


  This file has been truncated. show original

  
  
    
    
  
  


The specific error “No module named onnx” and the resolutions are available in this project. You can jump to this specific Jupyter notebook 24 and find your answer there.
I hope that is useful.