Breaking

2019년 3월 20일 수요일

ResourceExhaustedError: OOM when allocating tensor with shape 관련 error 해결법

ResourceExhaustedError: OOM when allocating tensor with shape 관련 error 해결법

https://github.com/tensorflow/tensorflow/issues/14724
https://github.com/tensorflow/models/issues/3393
https://github.com/tensorflow/tensorflow/issues/136

이 세 곳의 github에서 error점에 대한 이야기가 오갔고 이에 대한 해결법을 간결하게 요약하여 설명하자면

해결법 1. 모든 내용에 대한 것을 저장한 후 jupyter notebook을 다시 시작한다.

해결법 2. 말그대로 Memory 자원의 부족으로 생긴 문제이기 때문에 batch_size을 줄여준다. 일반적으로 batch_size는 $2^n$에 해당하는 양만큼이니 지수를 하나 줄여보는 것이 어떤가에 대한 제시를 한다.

해결법 3. 다른 GPU Memory 사용에 대한 소스를 없앤다.
더 쉽게 설명하자면 사용하는 프로그램 이외에 다 종료해서 Memory 사용을 줄이라는 말이다.

이 정도면 해결?