解决：“NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ back......“

出现问题的原因：

torch、torchvision、torchaudio版本不一致，

解决方法：卸载，重新安装

卸载：

pip uninstall torch torchvision torchaudio

重新安装

pip install torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 --index-url https://download.pytorch.org/whl/cu124