jackroos/VL-BERT

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

nikhilbchilwant opened this issue · 0 comments

nikhilbchilwant commented

Posting to help others

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

I resolved it by re-installing apex package on that GPU as below

git reset --hard 44532b30a4fad442f00635a0f4c8f241b06c2315
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./