람쥐썬더

tensorflow 학습 중 자주 보게 되는 OOM (Out Of Memory) 에러 .. 내부 모듈이 아닌 역전파로 학습 시키려다 보니까 어김없이 나타나고 말았다 뭐 하는것도 아닌데 메모리 드립다 먹고 있는 녀석 사실 한두번 이러면 그냥 재부팅 해주면 되는 문제긴 한데 계속 재부팅하기 귀찮아서 코드로 안되나 찾아봤음 Linux 환경이면 sudo kill 로 메모리를 죽일 수 있다는데 코드에서 해결 할 수 없나 찾아봤고 역시나 스택오버플로우에 의인들이 있었다 재부팅은 아니더라도 런타임은 다시 실행시켜줘야댐 .............. 해결 : pip install numba from numba import cuda device = cuda.get_current_device() device.reset() 조와