Python Library/Machine Learning
CUDA error: no kernel image is available for execution on the device 에러 해결
hellobird
2021. 5. 18. 12:25
https://github.com/NVIDIA/apex/issues/605
FusedLayerNorm leads to RuntimeError: CUDA error: no kernel image is available for execution on the device · Issue #605 · NVID
After a GPU tensor goes through FusedLayerNorm, the next time that memory is accessed I get a RuntimeError: CUDA error: no kernel image is available for execution on the device. To reproduce: impor...
github.com
Recent changes to Pytorch's built-in extension builder sometimes lead to it compiling for a different architecture.
라고함.
export TORCH_CUDA_ARCH_LIST="6.0;6.1;6.2;7.0;7.5"
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./