ModuleNotFoundError: No module named ‘torchvision.transforms.functional_tensor‘
出现ModuleNotFoundError: No module named 'torchvision.transforms.functional_tensor'的原因大概是原先的“名字”改了,但是安装的basicsr包中的名字没有改,所以会报错。改成from torchvision.transforms._functional_tensor import rgb_to_grayscale。关于我
问题
在进行深度学习训练过程中出现ModuleNotFoundError: No module named 'torchvision.transforms.functional_tensor'报错,多方查阅资料后得到了解决方案。
关于我的环境:CUDA==12.1 torch==2.4.1 GPU==4060
原因及解决方案
出现ModuleNotFoundError: No module named 'torchvision.transforms.functional_tensor'的原因大概是原先的“名字”改了,但是安装的basicsr包中的名字没有改,所以会报错。
只要在miniconda3/lib/python3.12/site-packages/basicsr/data/degradations.py文件中第8行将
原from torchvision.transforms.functional_tensor import rgb_to_grayscale
改成from torchvision.transforms._functional_tensor import rgb_to_grayscale
或者改成from torchvision.transforms.functional import rgb_to_grayscale
均能够解决问题。
更多推荐
所有评论(0)