Hint : If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info. This isn't available when running in Eager mode.

GPU Memory가 부족하다는 의미

 

해결 방안으로는 3가지 정도가 있다.

1. 데이터셋 크기 줄이기

2. 배치 사이즈 줄이기

3. 충분한 메모리 사용

 

주의 사항

ec2 서버를 사용할 경우 큰 인스턴스를 사용하면 되지만 요금 주의!!