PyTorch: RuntimeError: device >= 0 && device < num_gpus INTERNAL ASSERT FAILED #126344
Labels
module: cuda
Related to torch.cuda, and CUDA support in general
triaged
This issue has been looked at a team member, and triaged and prioritized into an appropriate module
When I'm trying to use PyTorch with YOLO with
yolo detect ... device=0
, I'm getting this error:RuntimeError: device >= 0 && device < num_gpus INTERNAL ASSERT FAILED at "../aten/src/ATen/cuda/CUDAContext.cpp":50, please report a bug to PyTorch. device=, num_gpus=
Outputs of PyTorch usage and GPU availability as shown below looks good though.
What is the problem? How to fix it?
Full stack trace:
cc @ptrblck @msaroufim
The text was updated successfully, but these errors were encountered: