memory
-
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..
-
TensorFlow GPU Memory error인공지능(AI)/TensorFlow 2020. 3. 9. 17:55
텐서플로우를 이용하여 GPU에서 트레이닝이나 추론을 할 때 메모리 에러를 만났다. 매 번 어려움을 겪어서 조금 정리해 본다. 에러 환경 : TensorFlow, GPU, CNN 을 조합으로 사용할 때 매 번 에러를 만났던 것 같다. 원인은 대부분 Out Of Memory 문제 였던 것 같은 데, 아래 링크에 비교적 잘 정리가 되어 있으니 참고. https://datamasters.co.kr/33 https://goodtogreate.tistory.com/entry/TensorFlow%EB%A5%BC-%EA%B3%B5%EC%9A%A9-GPU%EC%97%90%EC%84%9C-%EC%82%AC%EC%9A%A9-%ED%95%A0-%EB%95%8C-%EB%A9%94%EB%AA%A8%EB%A6%AC-%EC%A0%8..