No module named 'torch.onnx' when following super_resolution_with_caffe2.html

[liqunfu]

I am following tutorial http://pytorch.org/tutorials/advanced/super_resolution_with_caffe2.html (Transfering a model from PyTorch to Caffe2 and Mobile using ONNX). At the beginning I get:
ModuleNotFoundError Traceback (most recent call last)
in ()
5 from torch.autograd import Variable
6 import torch.utils.model_zoo as model_zoo
----> 7 import torch.onnx

ModuleNotFoundError: No module named 'torch.onnx'

My environment is: Ubuntu 16.04, anaconda 4.3.25. my environment has Python 3.6. PyTorch 0.20. onnx 0.1, torchvision 0.1.9.

Please let me know what is missing, Thanks,

[soumith]

for the onnx tutorial, you will have to install pytorch from source.
It is not part of a pytorch release yet:
https://github.com/pytorch/pytorch#from-source

[liqunfu]

Thanks @soumith! It worked with install pytorch from source.

[liuliuxiansheng]

@soumith hello，I also encounter this problem. But when I follow the official link, it doesn't work. And the error is :
Traceback (most recent call last):
File "", line 1, in
File "torch/init.py", line 13, in
from .version import version
ImportError: No module named version
Could you help me? How can I build pytorch from source ?

[chsasank]

You need to install build pytorch from source as described here: https://github.com/pytorch/pytorch#from-source