AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘

我在运行yolov5 6.0版本的detect.py时遇到该错误,后经调研发现是pytorch版本问题,当我把pytorch1.13版本换到2.2.0版本时,再运行detect.py时没有报错,正常运行。

以下是pytorch2.2.0版本的安装指令:

pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cu118

 在更换版本时要注意cuda版本是否匹配!!!

若cuda版本不匹配请更换cuda版本。

CUDA Toolkit Archive | NVIDIA Developer 

posted @ 2026-01-09 14:17  滑板拉半挂  阅读(2)  评论(0)    收藏  举报