ttzhrabc的博客

原创 NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend.

在成功安装torch2.1.1+cu118后运行YOLOv8程序，测试发现能够识别出torch版本及GPU，但运行YOLO程序时出现如题报错，尝试安装CPU版本发现能正常运行。最终发现应该是python版本过高导致出错，原始版本为3.11，修改为3.8后能够正常运行YOLO程序。记录一下为YOLOv8安装GPU版torch过程中遇到的问题。