ModuleNotFoundError: No module named ‘torchvision.transforms.functional_tensor‘

问题：

Traceback (most recent call last):
  File "train.py", line 3, in <module>
    from ultralytics import YOLO
  File "/home/pod/shared-nvme/ultralytics/__init__.py", line 5, in <module>
    from ultralytics.data.explorer.explorer import Explorer
  File "/home/pod/shared-nvme/ultralytics/data/__init__.py", line 3, in <module>
    from .base import BaseDataset
  File "/home/pod/shared-nvme/ultralytics/data/base.py", line 17, in <module>
    from ultralytics.data.utils import FORMATS_HELP_MSG, HELP_URL, IMG_FORMATS
  File "/home/pod/shared-nvme/ultralytics/data/utils.py", line 19, in <module>
    from ultralytics.nn.autobackend import check_class_names
  File "/home/pod/shared-nvme/ultralytics/nn/__init__.py", line 3, in <module>
    from .tasks import (
  File "/home/pod/shared-nvme/ultralytics/nn/tasks.py", line 2, in <module>
    from .Addmodules import *
  File "/home/pod/shared-nvme/ultralytics/nn/Addmodules/__init__.py", line 31, in <module>
    from .HAttention import *
  File "/home/pod/shared-nvme/ultralytics/nn/Addmodules/HAttention.py", line 5, in <module>
    from basicsr.archs.arch_util import to_2tuple, trunc_normal_
  File "/home/pod/shared-nvme/conda/envs/v8/lib/python3.8/site-packages/basicsr/__init__.py", line 4, in <module>
    from .data import *
  File "/home/pod/shared-nvme/conda/envs/v8/lib/python3.8/site-packages/basicsr/data/__init__.py", line 22, in <module>
    _dataset_modules = [importlib.import_module(f'basicsr.data.{file_name}') for file_name in dataset_filenames]
  File "/home/pod/shared-nvme/conda/envs/v8/lib/python3.8/site-packages/basicsr/data/__init__.py", line 22, in <listcomp>
    _dataset_modules = [importlib.import_module(f'basicsr.data.{file_name}') for file_name in dataset_filenames]
  File "/home/pod/shared-nvme/conda/envs/v8/lib/python3.8/importlib/__init__.py", line 127, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
  File "/home/pod/shared-nvme/conda/envs/v8/lib/python3.8/site-packages/basicsr/data/realesrgan_dataset.py", line 11, in <module>
    from basicsr.data.degradations import circular_lowpass_kernel, random_mixed_kernels
  File "/home/pod/shared-nvme/conda/envs/v8/lib/python3.8/site-packages/basicsr/data/degradations.py", line 8, in <module>
    from torchvision.transforms.functional_tensor import rgb_to_grayscale
ModuleNotFoundError: No module named 'torchvision.transforms.functional_tensor'

解决方案：

conda文件夹下

/conda/envs/v8/lib/python3.8/site-packages/basicsr/data/degradations.py

from torchvision.transforms.functional_tensor import rgb_to_grayscale
改成

from torchvision.transforms.functional import rgb_to_grayscale

就行啦

原文

深度学习No module named ‘torchvision.transforms.functional_tensor‘问题解决_modulenotfounderror:no module named 'torchvision.t-CSDN博客