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()
'# Machine Learning > 글 공부' 카테고리의 다른 글
BERT 설명하기 (3) | 2020.08.07 |
---|---|
클래스 불균형, UnderSampling & OverSampling (0) | 2019.09.06 |
Tensorflow 특정 gpu 사용하기 (0) | 2019.07.22 |
(Ubuntu) Ndivia driver 삭제, cuda 삭제 (0) | 2019.07.11 |
(Neural Network) train 후 validation data를 어떻게 처리해야 할까? (0) | 2019.07.02 |