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를 입력해주어서 발생한 오류였다.
연구는 디버그의 연속이다.
디버그는 계속 발생하고 하루 종일 디버그만 해결하다가 하루가 끝나네
내 잃어버린 하루
반응형