반응형 PyTorch GPU 메모리 해제1 [MLOps] PyTorch GPU Out-Of-Memory 문제 해결 PyTorch를 이용하여 ML Pipeline을 개발하던 중 특정 시간에 학습을 시킬 때 메모리가 꽉 차서 모델 학습할 때 실패하는 경우가 있었다. ML Pipeline을 구성할 때 메모리도 넉넉하게 사용할 수 있도록 구성하지만 의도치 않게 OOM문제가 발생하는 경우가 있었다. APScheduler를 이용해서 구성했는데 학습을 하면서 out-of-memory 이슈가 발생해서 문제를 해결하기 위해 찾아보았고, 같은 이유로 해결방법을 찾는 사람들에게 도움이 되고자 한다. 파이토치 뿐만 아니라 텐서플로우를 사용하더라도 마찬가지이므로 도움이 될 것 이다. GPU Out-Of-Memory 발생 RuntimeError: CUDA out of memory. Tried to allocate 126.00 MiB (GPU.. 2021. 7. 17. 이전 1 다음 반응형