nat_pagle 2023-03-07 13:48 采纳率: 66.7%
浏览 648
已结题

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

运行yolov5工程时出现问题

img

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

  • 写回答

4条回答 默认 最新

  • 关注

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

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(3条)

报告相同问题?

问题事件

  • 系统已结题 3月31日
  • 已采纳回答 3月23日
  • 创建了问题 3月7日