Cannot import name 'QuantStub' from 'torch.ao.quantization'

quantization
1

Hi I am new to pytorch.

I encounter an issue when I run the below. Can someone assist ?
from torchvision import datasets, transforms

Error message:
ImportError: cannot import name ‘QuantStub’ from ‘torch.ao.quantization’

2

Are you seeing this error while just running from torchvision import datasets, transforms or any other imports?
Could you post the versions reported by pip list | grep torch and/or conda list | grep torch?

3

Hi ptrblck,
Thanks for reply. Someone suggested that I take away the .ao from all the error message files. So i just took away .ao from the below files and now there is no error message. Hope it really resolve the issue.

init
mobilenetv2
mobilenetv3

1 Like
4

you probably have an older version of pytorch? we are deprecating the torch.quantization namespace, and torch.ao.quantization is the recommended one

1 Like
5

Hi Jerry,
Oh I see. May I know what version of pytorch & python should I install so that I can run pytorch smoothly?

2 Likes
6

Updating pytorch fixes the problem:

pip install --upgrade torch torchvision