是torchvisiao-cuda沒裝好，重裝下

直接：

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple

參考文章：解決“NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend.”

總結

以上是生活随笔為你收集整理的YOLOv5 报错：“NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ back的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。