ModuleNotFoundError: No module named 'torch.ao'

[brace-great]

求助,跟着教程走到执行train.sh报错:

/root/miniconda3/envs/pixivai/lib/python3.9/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension: /root/miniconda3/envs/pixivai/lib/python3.9/site-packages/torchvision/image.so: undefined symbol: _ZNK3c106IValue23reportToTensorTypeErrorEv
  warn(f"Failed to load image Python extension: {e}")
Traceback (most recent call last):
  File "/root/pixiv_AI_crawler/main.py", line 21, in <module>
    from timm.data.mixup import Mixup
  File "/root/miniconda3/envs/pixivai/lib/python3.9/site-packages/timm/__init__.py", line 2, in <module>
    from .models import create_model, list_models, is_model, list_modules, model_entrypoint, \
  File "/root/miniconda3/envs/pixivai/lib/python3.9/site-packages/timm/models/__init__.py", line 1, in <module>
    from .cspnet import *
  File "/root/miniconda3/envs/pixivai/lib/python3.9/site-packages/timm/models/cspnet.py", line 19, in <module>
    from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
  File "/root/miniconda3/envs/pixivai/lib/python3.9/site-packages/timm/data/__init__.py", line 4, in <module>
    from .transforms import *
  File "/root/miniconda3/envs/pixivai/lib/python3.9/site-packages/timm/data/transforms.py", line 2, in <module>
    import torchvision.transforms.functional as F
  File "/root/miniconda3/envs/pixivai/lib/python3.9/site-packages/torchvision/__init__.py", line 7, in <module>
    from torchvision import models
  File "/root/miniconda3/envs/pixivai/lib/python3.9/site-packages/torchvision/models/__init__.py", line 18, in <module>
    from . import quantization
  File "/root/miniconda3/envs/pixivai/lib/python3.9/site-packages/torchvision/models/quantization/__init__.py", line 3, in <module>
    from .mobilenet import *
  File "/root/miniconda3/envs/pixivai/lib/python3.9/site-packages/torchvision/models/quantization/mobilenet.py", line 1, in <module>
    from .mobilenetv2 import *  # noqa: F401, F403
  File "/root/miniconda3/envs/pixivai/lib/python3.9/site-packages/torchvision/models/quantization/mobilenetv2.py", line 6, in <module>
    from torch.ao.quantization import QuantStub, DeQuantStub
ModuleNotFoundError: No module named 'torch.ao'
Killing subprocess 7044
Traceback (most recent call last):
  File "/root/miniconda3/envs/pixivai/lib/python3.9/runpy.py", line 197, in _run_module_as_main
    return _run_code(code, main_globals, None,
  File "/root/miniconda3/envs/pixivai/lib/python3.9/runpy.py", line 87, in _run_code
    exec(code, run_globals)
  File "/root/miniconda3/envs/pixivai/lib/python3.9/site-packages/torch/distributed/launch.py", line 340, in <module>
    main()
  File "/root/miniconda3/envs/pixivai/lib/python3.9/site-packages/torch/distributed/launch.py", line 326, in main
    sigkill_handler(signal.SIGTERM, None)  # not coming back
  File "/root/miniconda3/envs/pixivai/lib/python3.9/site-packages/torch/distributed/launch.py", line 301, in sigkill_handler
    raise subprocess.CalledProcessError(returncode=last_return_code, cmd=cmd)
subprocess.CalledProcessError: Command '['/root/miniconda3/envs/pixivai/bin/python', '-u', 'main.py', '--local_rank=0', '--model', 'convnext_tiny', '--drop_path', '0.5', '--input_size', '384', '--batch_size', '36', '--lr', '2e-4', '--update_freq', '2', '--nb_classes', '2', '--warmup_epochs', '0', '--epochs', '60', '--weight_decay', '1e-8', '--head_init_scale', '0.001', '--cutmix', '0', '--mixup', '0', '--save_ckpt_num', '100', '--dist_eval', 'False', '--color_jitter', '0.2', '--finetune', 'ckpt/checkpoint-best_t5.pth', '--data_path', '/dataset/dzy/anime_ds', '--eval_data_path', '/dataset/dzy/anime_ds', '--imset', './imset/', '--data_set', 'json', '--output_dir', 'output/tiny6']' returned non-zero exit status 1.

---