CUDA out of memory (계속 추가)

𝐌𝐢𝐧𝐣𝐢·2021년 7월 1일
0

오류해결

목록 보기
2/6
post-custom-banner

[Problem]

RuntimeError: CUDA out of memory

[Solution]

1) batch size를 줄인다 128->64->32...
2)

import torch, gc
gc.collect()
torch.cuda.empty_cache()

3) (우분투 이용시 가장 효과가 좋았던...)
커널 종료 후 다시 실행 --> 미처 종료되지 않았던 python들이 종료되는듯?

0개의 댓글