AttributeError: ‘Upsample‘ object has no attribute ‘recompute

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

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