개발 이야기/Machine learning
Keras에서 GPU 메모리 할당량 조절
가마뫼
2016. 11. 18. 11:28
TensorFlow를 Backend로 사용하는 GPU Enabled Keras는 기본값이 모든 GPU Memory를 할당하는 것이다.
혼자 쓰는 것이면 문제가 안 되겠지만, 연구실 구성원들과 같이 쓰는 서버이기 때문에
메모리 할당량을 조절하려고 검색을 하니 다음 두 링크를 찾을 수 있었다.
https://groups.google.com/forum/#!topic/keras-users/MFUEY9P1sc8