[PYTHON] GPU VRAM(메모리) 초기화 (Winodows)

tensorflow 학습 중 자주 보게 되는 OOM (Out Of Memory) 에러 ..

 

내부 모듈이 아닌 역전파로 학습 시키려다 보니까 어김없이 나타나고 말았다

 

뭐 하는것도 아닌데 메모리 드립다 먹고 있는 녀석

 

사실 한두번 이러면 그냥 재부팅 해주면 되는 문제긴 한데 계속 재부팅하기 귀찮아서 코드로 안되나 찾아봤음

 

Linux 환경이면 sudo kill 로 메모리를 죽일 수 있다는데 코드에서 해결 할 수 없나 찾아봤고 역시나 스택오버플로우에 의인들이 있었다 재부팅은 아니더라도 런타임은 다시 실행시켜줘야댐 ..............

 

 

해결 : 

pip install numba

from numba import cuda 

device = cuda.get_current_device()
device.reset()

조와