我在运行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