报错解决——ModuleNotFoundError: No module named ‘torch.ao‘

完整报错：

Traceback (most recent call last):
  File "test.py", line 12, in <module>
    from torchvision.utils import save_image
  File "/mnt/A/hust_csj/anaconda3/envs/iKUN/lib/python3.8/site-packages/torchvision/__init__.py", line 6, in <module>
    from torchvision import datasets, io, models, ops, transforms, utils
  File "/mnt/A/hust_csj/anaconda3/envs/iKUN/lib/python3.8/site-packages/torchvision/models/__init__.py", line 17, in <module>
    from . import detection, optical_flow, quantization, segmentation, video
  File "/mnt/A/hust_csj/anaconda3/envs/iKUN/lib/python3.8/site-packages/torchvision/models/quantization/__init__.py", line 3, in <module>
    from .mobilenet import *
  File "/mnt/A/hust_csj/anaconda3/envs/iKUN/lib/python3.8/site-packages/torchvision/models/quantization/mobilenet.py", line 2, in <module>
    from .mobilenetv3 import *  # noqa: F401, F403
  File "/mnt/A/hust_csj/anaconda3/envs/iKUN/lib/python3.8/site-packages/torchvision/models/quantization/mobilenetv3.py", line 6, in <module>
    from torch.ao.quantization import DeQuantStub, QuantStub
ModuleNotFoundError: No module named 'torch.ao'

解决办法：

将报错文件 mobilenetv3.py 中的

from torch.ao.quantization import DeQuantStub, QuantStub

修改为

from torch.quantization import DeQuantStub, QuantStub