Error "Could not run 'torchvision::nms' with arguments from the 'CUDA' backend" in detect.py with cuda

[developer-119]

Hi this error happend when run detect.py:

File "E:\ANPRir\yolov7_env\lib\site-packages\torchvision\ops\boxes.py", line 40, in nms return torch.ops.torchvision.nms(boxes, scores, iou_threshold) 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 during 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 resolutions. 'torchvision::nms' is only available for these backends: [CPU, QuantizedCPU, BackendSelect, Python, Named, Conjugate, Negative, ZeroTensor, ADInplaceOrView, AutogradOther, AutogradCPU, AutogradCUDA, AutogradXLA, AutogradLazy, AutogradXPU, AutogradMLC, AutogradHPU, Tracer, AutocastCPU, Autocast, Batched, VmapMode, Functionalize].

Instalation command: pip install install torch==1.11.0+cu113 torchvision==0.12.0 torchaudio==0.11.0 --find-links ..\torch-1.11.0+cu113-cp38-cp38-win_amd64.whl

Environment: Windows 10 python 3.8 gtx 1050 ti

[ninadchobe]

Hey, I'm facing the same issue. What steps you took to overcome that ? I'm using cuda 12.1 and have installed compatible tochvision and torch.

[developer-119]

Hi, I finally realized this problem in my selected packages(torchvision does not match with torch), and after some changes the problem was solved.

pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118

you should try with your own system and python version, i used this: python 3.8 torch 2.3.1+cu118 torchaudio 2.3.1+cu118 torchvision 0.18.1+cu118

[developer-119]

Hi, I finally realized this problem in my selected packages(torchvision does not match with torch), and after some changes the problem was solved.

pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118

you should try with your own system and python version, i used this: python 3.8 torch 2.3.1+cu118 torchaudio 2.3.1+cu118 torchvision 0.18.1+cu118

On Mon, Jul 29, 2024 at 4:45 PM Ninad Chobe @.***> wrote:

Hey, I'm facing the same issue. What steps you took to overcome that ? I'm using cuda 12.1 and have installed compatible tochvision and torch.

— Reply to this email directly, view it on GitHub https://github.com/WongKinYiu/yolov7/issues/1655#issuecomment-2255912381, or unsubscribe https://github.com/notifications/unsubscribe-auth/AV2HUELVJPKJ42KEYQVM3ITZOY567AVCNFSM6AAAAABLUIOFJOVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZDENJVHEYTEMZYGE . You are receiving this because you authored the thread.Message ID: @.***>