본문 바로가기

Pytorch/Errors

[Pytorch 오류] Pytorch Dataparallel시 CUDA_LAUNCH_BLOCKING=1 현상.

반응형

다양한 이유가 있을 수 있겠지만, 나같은 경우 imagenet의 validation dataset이 올바르게 설정되어 있지 않아서였다.

validation dataset을 다시 다운받은 후, 디렉토리를 설정해주니 정상 작동했다.

ImageNet dataset 설치 방법에 대한 글을 작성한 후, 글을 첨부하도록 하겠다.

반응형