torch与torchvision版本对应关系 & ImportError: cannot import name ‘QuantStub‘ from ‘torch.ao.quantization‘
torch与torchvision版本对应关系看下边的表格
兄弟们看看是不是报这个：
File "xxxxxxxx/xxxxx/tools/train.py", line 10, in <module>
from mmcv.runner import init_dist
File "xxxxxxxx/.conda/envs/ga/lib/python3.7/site-packages/mmcv/runner/__init__.py", line 2, in <module>
from .checkpoint import (_load_checkpoint, load_checkpoint, load_state_dict,
File "xxxxxxxx/.conda/envs/ga/lib/python3.7/site-packages/mmcv/runner/checkpoint.py", line 11, in <module>
import torchvision
File "xxxxxxxx/.conda/envs/ga/lib/python3.7/site-packages/torchvision/__init__.py", line 7, in <module>
from torchvision import models
File "xxxxxxxx/.conda/envs/ga/lib/python3.7/site-packages/torchvision/models/__init__.py", line 18, in <module>
from . import quantization
File "xxxxxxxx/.conda/envs/ga/lib/python3.7/site-packages/torchvision/models/quantization/__init__.py", line 3, in <module>
from .mobilenet import *
File "xxxxxxxx/.conda/envs/ga/lib/python3.7/site-packages/torchvision/models/quantization/mobilenet.py", line 1, in <module>
from .mobilenetv2 import * # noqa: F401, F403
File "xxxxxxxx/.conda/envs/ga/lib/python3.7/site-packages/torchvision/models/quantization/mobilenetv2.py", line 6, in <module>
from torch.ao.quantization import QuantStub, DeQuantStub
ImportError: cannot import name 'QuantStub' from 'torch.ao.quantization'
参考自：
https://forums.fast.ai/t/cannot-import-name-quantstub-from-torch-ao-quantization/94078/2
意思是可能 torch 和 torchvision 的版本不匹配
看下自己的版本
>>> import torch
>>> torch.__version__
'1.10.0+cu113'
这样说明是在
https://download.pytorch.org/whl/torch_stable.html
下载的，如果没有 cu113 大概率pip或者conda下载的
- 如果没有
cu113就直接参考下边的表格，pip install torchvision==对应的版本即可
| torch | torchvision | torchaudio |
|---|---|---|
| pytorch==1.12.0 | torchvision==0.13.0 | torchaudio==0.12.0 |
| pytorch==1.11.0 | torchvision==0.12.0 | torchaudio==0.11.0 |
| pytorch==1.10.1 | torchvision==0.11.2 | torchaudio==0.10.1 |
| pytorch==1.10.0 | torchvision==0.11.0 | torchaudio==0.10.0 |
| pytorch==1.9.1 | torchvision==0.10.1 | torchaudio==0.9.1 |
| pytorch==1.9.0 | torchvision==0.10.0 | torchaudio==0.9.0 |
| pytorch==1.8.1 | torchvision==0.9.1 | torchaudio==0.8.1 |
| pytorch==1.8.0 | torchvision==0.9.0 | torchaudio==0.8.0 |
| pytorch==1.7.1 | torchvision==0.8.2 | torchaudio==0.7.2 |
| pytorch==1.7.0 | torchvision==0.8.0 | torchaudio==0.7.0 |
| pytorch==1.6.0 | torchvision==0.7.0 | |
| pytorch==1.5.1 | torchvision==0.6.1 | |
| pytorch==1.5.0 | torchvision==0.6.0 | |
| pytorch==1.4.0 | torchvision==0.5.0 | |
| pytorch==1.2.0 | torchvision==0.4.0 | |
| pytorch==1.1.0 | torchvision==0.3.0 |
表格参考自：
https://pytorch.org/get-started/previous-versions/
- 如果后边带着
cu113之类指定 cuda 版本的，那只能从这个网站下载了
https://download.pytorch.org/whl/torch_stable.html
比如我是 cuda11.2，那就找大于它但是最接近的版本，以及指定python版本，同时根据上述表格，找到torch1.10对应的torchvision版本0.11
下载这个 wheel 文件，pip 安装即可
如果不这样搞，直接 pip install torchvision==0.11 会报这个错：
Collecting torchvision==0.11
Using cached torchvision-0.11.0-cp37-cp37m-manylinux1_x86_64.whl (23.3 MB)
Requirement already satisfied: numpy in /xxxxx/.conda/envs/ga/lib/python3.7/site-packages (from torchvision==0.11) (1.21.5)
ERROR: Could not find a version that satisfies the requirement torch==1.10.0+cu102 (from torchvision) (from versions: 1.0.0, 1.0.1, 1.0.1.post2, 1.1.0, 1.2.0, 1.3.0, 1.3.1, 1.4.0, 1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2, 1.11.0, 1.12.0, 1.12.1)
ERROR: No matching distribution found for torch==1.10.0+cu102