How to solve this error? RuntimeError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 17.9k
- Forks
- 7.3k
- Avg merge
- 1d 15h
- Merged PRs (30d)
- 13
Description
❓ Questions and Help
Please note that this issue tracker is not a help form and this issue will be closed.
I'm beginner of ML and trying to use some solution based on pytorch (called detectron2)
When the solution inferred the image, I always got the below error.
RuntimeError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend. 'torchvision::nms' is only available for these backends: [CPU, BackendSelect, Named, AutogradOther, AutogradCPU, AutogradCUDA, AutogradXLA, Tracer, Autocast, Batched, VmapMode].
Actually, I didn't get this error and couldn't search anything about this on google.
Is there anybody who knows the way to handle this?
Info:
I installed the CUDA v11.1 from https://developer.nvidia.com/cuda-downloads
torch version: 1.7.0
torchvision version: 0.8.0
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
No repository file, test, or entry point is named. Start by reproducing the error with torch 1.7.0, torchvision 0.8.0, CUDA 11.1, and detectron2; completion would require a confirmed cause and an actionable resolution.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- computer-vision, machine-learning
- Issue type
- Bug
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 5/100