pytorch 1.8.0 램 메모리 누수 현상 발견 및 해결
2021. 12. 18. 19:19ㆍ분석 Python/Pytorch
실험을 하다가 램을 확인하는데 엄청나게 많이 쌓이는 것을 확인하였다
memory_profiler로도 확인을 해보니 forward를 할 때마다 램이 계속 쌓이는 현상이 발견되서 이것 저것 실험을 해도 계속 같은 증상이 나왔다.
그래서 마지막으로 패키지 버전은 올려봤다
현재 21.12 월 기준 최신버전인 1.10.1+cpu 버전으로 진행하니 해결되었다.
앞으로 이런 버그가 있는 경우 최후의 수단으로 패키지를 업그레이드 혹은 다운그레이드 해보는 것도 좋아보인다
끝
728x90
'분석 Python > Pytorch' 카테고리의 다른 글
| Pytorch 1.11 이후) functorch 알아보기 (0) | 2022.03.14 |
|---|---|
| Pytorch 1.11 이후) torchdata 알아보기 (0) | 2022.03.14 |
| pytorch) dataloader sampler (4) | 2021.04.19 |
| [Pytorch] gather 함수 설명 (특정 인덱스만 추출하기) (1) | 2021.03.17 |
| PyGAD + Pytorch + Skorch+ torch jit (0) | 2021.01.30 |