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版本。

浙公网安备 33010602011771号