Python/PyTorch 공부

[PyTorch] RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR

AI 꿈나무 2021. 12. 3. 01:12
반응형

/opt/conda/conda-bld/pytorch_1634272068694/work/aten/src/ATen/native/cuda/Indexing.cu:699: indexSelectLargeIndex: block: [132,0,0], thread: [31,0,0] Assertion `srcIndex < srcSelectDimSize` failed.

RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR

 

위와 같은 에러가 발생했다.

 

 nn.embedding으로 생성한 embedding vector의 차원이 [15,600]인데

 

 15가 넘는 index를 입력해주어서 발생한 오류였다.

 

 연구는 디버그의 연속이다.

 

 디버그는 계속 발생하고 하루 종일 디버그만 해결하다가 하루가 끝나네

 

 내 잃어버린 하루

 

 

반응형