[해결실패] GPU 메모리부족: RuntimeError: Dst tensor is not initialized

주피터 노트북에서 실행시키다가 runtime error가 발생했다.

방금 전까지는 잘됐는데 갑자기 Dst tensor is not initialized 란다.

 

 

 

원인은 메모리 부족.

TensorFlow GPU Memory error - 절대적으로 GPU 메모리가 부족할 때 나는 에러다.

 

 

 

 

해결 방법은 dataset 크기를 줄이기

train의 경우 batch size, datasize를 줄일 수 있고

inference의 경우 datasize를 줄일 수 있겠다

 

 

 

본인은 inference의 경우였으나 datasize를 줄일 수 없는 경우.

해결 못하고 cpu로 우선 실행했다

 

 

 

 

혹시 아는 사람이 계시다면 댓글 남겨주세요