RuntimeError: CUDA error: no kernel image is available for execution on the device

HRNet / HigherHRNet-Human-Pose-Estimation

This is an official implementation of our CVPR 2020 paper "HigherHRNet: Scale-Aware Representation Learning for Bottom-Up Human Pose Estimation" (https://arxiv.org/abs/1908.10357)

MIT License

1.35k stars 272 forks source link

RuntimeError: CUDA error: no kernel image is available for execution on the device #95

Closed salvador-blanco closed 2 years ago

salvador-blanco commented 2 years ago

RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.

I was having this problem, but by running

pip3 install torch==1.7.1+cu110 torchvision==0.8.2+cu110 -f https://download.pytorch.org/whl/torch_stable.html

I was able to fix it

salvador-blanco commented 2 years ago

https://github.com/pytorch/pytorch/issues/49161