Tensorflow object detection api 에러, keras에러 - failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED, could not create cudnn handle: CUDNN_STATU_ALLOC_FAILED)

텐서플로우 오브젝트 디텍션 에이피아이(Tensorflow object detection api) 를 사용하다가 제목과 같은 문제가 발생하였습니다.

 

에러내용은 아래와 같습니다 : 

failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED, could not create cudnn handle: CUDNN_STATU_ALLOC_FAILED

 

구글링을 아무리 해봐도 해결책이 안나왔는데 어떤 중국인이 해결방법을 적어놨습니다. 중국어라서 그냥 넘어가려다가 번역이라도 해보자~ 하고 구글 번역을 돌렸는데 "해결책"이라는 단어가 있어서 집중하여 정독했습니다ㅋㅋ

 

문제의 원인은 GPU메모리 할당 문제라고 합니다.

해결책은 (import 아래 코드가 시작되는 가장 첫 줄에) 다음의 3줄을 입력해주면 됩니다.

 

 

이걸로 몇시간동안 고생했네요 ㅠㅠ 다들 화이팅! 댓글, 좋아요 부탁드려요~~ : )