텐설플로
-
TensorFlow GPU Memory error - 절대적으로 GPU 메모리가 부족할 때 나는 에러인공지능(AI)/TensorFlow 2020. 3. 9. 18:02
이미지 사이즈가 큰 데이터로 딥러닝 트레이닝을 하려고 했는데 에러가 나고 안됐다. 절대적으로 GPU 메모리가 부족하다고 결론을 내렸다. 배치 사이즈가 1이므로 배치 크기를 더 줄일 수도 없었다. 이미지 크기를 잘게 잘라서 했더니 실행이 되었다. 아래와 같은 에러 메시지가 나왔었다. 에러 로그 메시지 일부------------------------------------------------------------- 아래처럼 별 에러 없이 메모리를 계속 할당하다가 결국 메모리가 부족해서 에러가 난다고 말하는 것 같은 메시지가 보였다. 2020-02-19 00:42:50.537629: I tensorflow/core/common_runtime/bfc_allocator.cc:674] 2 Chunks of size..