# Machine Learning/글 공부
keras, Model RAM 놓지 않는 현상
Hwiyong Jo
2019. 8. 20. 15:31
Model.fit 후에 세션이 종료되고 램을 놔주지 않는 현상이 있다.
* 케라스 내부적으로 해결이 되지 않은 이슈라는 글을 본적이 있다.
1. del model * 해결되지 않음
2. gc.collect() + gc.collect() 여러번 * 해결되지 않음
3. tf.as_default_graph() 안에서 처리 * 해결되지 않음
4. K.clear_session() * Case by Case
* 사용은 하고 있지만 죽는 경우도 있고, 안죽는 경우도 있었습니다. jupyter notebook의 경우 같은 cell 안에서 fit과 함께 실행시킨 경우에 될지도 모르겠네요.
5. 확실한 방법. + 이 방법과 비슷하게 프로세스를 죽이는 방법도 있음. nvidia-smi를 통해 PID를 확인한 후 kill
from numba import cuda
cuda.select_device(num)
cuda.close()