- 博客(1)
- 收藏
- 关注
原创 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版本可以比官方的版本高,可以不用强制要求一样。
2024-07-23 17:03:22
403
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人