NotImplementedError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend. This could be because the

这个报错意味着在使用PyTorch的torchvision库中的非极大值抑制（NMS）函数时，尝试在CUDA后端上运行该函数，但是该函数并未在CUDA后端上实现。这可能是因为该函数没有被编译到CUDA后端中，或者在自定义构建过程中被省略了。建议检查代码中是否正确地调用了NMS函数，并且确保在使用CUDA后端时，只使用已经在CUDA后端中实现的函数。如果需要在CUDA后端上使用NMS函数，则需要重新编译PyTorch并确保NMS函数被正确地编译到CUDA后端中。