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

jackroos / VL-BERT

Code for ICLR 2020 paper "VL-BERT: Pre-training of Generic Visual-Linguistic Representations".

MIT License

735 stars 110 forks source link

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

Closed nikhilbchilwant closed 3 years ago

nikhilbchilwant commented 3 years ago

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" ./