Pytorch Cuda error
-
[PyTorch Cuda 오류] cuda error: device-side assert triggeredProgramming Error/PyTorch 2021. 6. 6. 19:34
cuda error: device-side assert triggered PyTorch 뿐만 아니라 Tensorflow를 사용하다보면 위와 같은 에러를 자주 마주했을거라 생각합니다. 위 오류는 GPU를 사용하다보면 나올 수 있는 가장 추상적인(원인이 매우 많음) 에러입니다. 따라서, 위 에러만 보고서는 어떤 것에 문제가 생기는지 알기가 힘듭니다. 이를 해결하기 위해 우선, 해당 .py 파일 또는 Notebook 가장 위에 아래의 코드를 삽입합니다. import os os.environ['CUDA_LAUNCH_BLOCKING'] = "1" os.environ["CUDA_VISIBLE_DEVICES"] = "0" 그리고, 코드를 실행하면 조금 더 구체화된 에러 메시지가 나오게 됩니다. 제가 겪었던 경우 중 ..