m0_54041074的博客

原创 Could not run ‘torchvision::nms‘ .. from the ‘CUDA‘ backend 原因分析和解决

报错的直接原因是torchvision库中的文件： ... python311\Lib\site-packages\torchvision\ops\boxes.py 第41行，然后pip install torchvision-0.17.1+cu121-cp311-cp311-win_amd64.whl 即可。因此我下载的whl文件是 torchvision-0.17.1+cu121-cp311-cp311-win_amd64.whl。其中cuda版本可以比官方的版本高，可以不用强制要求一样。