Could not run 'torchvision::nms' with arguments from the 'CUDA' backend.有没有不动pytorch的解决方法

运行yolov5工程时出现问题

是pytorch版本和cuda版本不匹配么？有没有不动pytorch的解决方法，安pytorch费了很大劲

如果pytorch版本和cuda版本匹配，不用动pytorch

https://blog.csdn.net/qq_41590635/article/details/112384718

该回答引用GPTᴼᴾᴱᴺᴬᴵ
根据错误提示，这个问题可能是由于PyTorch版本和CUDA版本不兼容导致的。建议您检查一下PyTorch和CUDA的版本是否匹配。可以在PyTorch的官方网站上查看版本兼容性。如果版本不兼容，则需要升级或降级其中一个。另外，您可以尝试使用CPU来运行代码，以便解决与CUDA相关的问题。可以将torch.device('cuda')改为torch.device('cpu')来实现。但是，这样可能会降低代码的运行速度和性能。

这篇文章：Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend的解决；用GPU跑Detectron2的demo 也许能够解决你的问题，你可以看下

您好，我是有问必答小助手，您的问题已经有小伙伴帮您解答，感谢您对有问必答的支持与关注！
PS：问答VIP年卡【限时加赠：IT技术图书免费领】，了解详情>>> https://vip.csdn.net/askvip?utm_source=1146287632