完整错误：
NotImplementedError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend. This could be because the operator doesn't exist for this backend, or was omitted during the selective/custom build process (if using custom build). If you are a Facebook employee using PyTorch on mobile, please visit https://fburl.com/ptmfixes for possible resolutions. 'torchvision::nms' is only available for these backends: [CPU, QuantizedCPU, BackendSelect, Python, FuncTorchDynamicLayerBackMode, Functionalize, Named, Conjugate, Negative, ZeroTensor, ADInplaceOrView, AutogradOther, AutogradCPU, AutogradCUDA, AutogradXLA, AutogradMPS, AutogradXPU, AutogradHPU, AutogradLazy, Tracer, AutocastCPU, AutocastCUDA, FuncTorchBatched, FuncTorchVmapMode, Batched, VmapMode, FuncTorchGradWrapper, PythonTLSSnapshot, FuncTorchDynamicLayerFrontMode, PythonDispatcher].
我在 Windows 11 机器上尝试训练 YOLOv8 模型时得到这个，第一个时期一切正常，然后就发生了。
我在第一个纪元结束后也立即收到此错误，但我认为这无关紧要。
Error executing job with overrides: ['task=detect', 'mode=train', 'model=yolov8n.pt', 'data=custom.yaml', 'epochs=300', 'imgsz=160', 'workers=8', 'batch=4']
我正在尝试使用 CUDA GPU 训练 YOLOv8 图像检测模型。
当您安装了用于 CUDA 的 Torch 和 Torchaudio 但没有安装用于 CUDA 的 Torchvision 时，会发生此错误。
卸载Torch和Torchvision，我用的是pip：
pip uninstall torch torchvision
然后去这里使用漂亮的界面安装两者的正确版本。我得到以下命令：
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
我不知道是什么导致了这个错误，但我知道安装正确版本的 torchvision 和 cuda 并没有修复它。我解决的方法是卸载所有软件包，然后重新安装。现在工作得很好。事后看来，更新我所有的包可能更好，但问题还是解决了。