pytorch遇到的问题与解决办法

?Bunny

已于 2024-07-15 20:28:46 修改

阅读量148

点赞数 7

文章标签： pytorch 人工智能 python

于 2024-07-15 19:22:22 首次发布

本文链接：https://blog.csdn.net/qq_46229971/article/details/140446747

版权

pytorch遇到的问题与解决办法

yolo8 无法使用GPU运行，报错

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 duri
ng 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 resoluti
ons. 'torchvision::nms' is only available for these backends: [CPU, Meta, QuantizedCPU, BackendSelect, Python, FuncTorchDynamicLayerBackMode, Functionalize, Named, Conjugate, Negat
ive, ZeroTensor, ADInplaceOrView, AutogradOther, AutogradCPU, AutogradCUDA, AutogradXLA, AutogradMPS, AutogradXPU, AutogradHPU, AutogradLazy, AutogradMeta, Tracer, AutocastCPU, Aut
ocastCUDA, FuncTorchBatched, BatchedNestedTensor, FuncTorchVmapMode, Batched, VmapMode, FuncTorchGradWrapper, PythonTLSSnapshot, FuncTorchDynamicLayerFrontMode, PreDispatch, Python
Dispatcher].

报错原因：先安装GPU版本的pytorch和torchvision，再安装yolopip install ultralytics时，导致捆绑安装了cpu版本的torchvision并覆盖了原本的gpu版本的
解决办法：
a. 首先卸载CPU版本的torchvision pip3 uninstall torchvision
b. 官网下载torch版本对应的torchvision版本，并安装pip3 install F:\GOLGLE\torchvision-0.18.1+cu121-cp311-cp311-win_amd64.whl