728x90
RuntimeError: CUDA error: no kernel image is available for execution on the device
-
[Pytorch error] RuntimeError: CUDA error: no kernel image is available for execution on the device검색하기 귀찮아서 블로그에 박제 2022. 7. 1. 16:29
서버를 공유하고 있는데 누가 자꾸 가상환경에 환경설정을 안하나보다 ^^ 새학기마다 재설정하는 듯 pytorch 버전과 cuda 버전의 호환성 문제로 pytorch 재설치를 권장한다. 우선 내 환경은 linux이다. https://developer.nvidia.com/cuda-gpus CUDA GPUs - Compute Capability Explore your GPU compute capability and CUDA-enabled products. developer.nvidia.com 위 링크에서 우선 내 GPU의 compute capability를 확인한다. 나는 RTX A5000이므로 8.6이다. $ export TORCH_CUDA_ARCH_LIST=8.0 환경변수를 설정한다. # conda $ co..