CUDA out of memory 문제(파이토치)

Question

RuntimeError: CUDA out of memory. Tried to allocate 14.00 MiB (GPU 0; 3.81 GiB total capacity; 2.09 GiB already allocated; 40.19 MiB free; 2.11 GiB reserved in total by PyTorch) GPU가 많이 모자란데 해결방법이 따로 있을까요? 데이터를 분할 해서 학습 시키면 된다고 하는데 하는 방법을 찾아봐도 이해할 수가 없습니다ㅠ

 · Answer

batch size를 줄이시거나 메모리를 적게 사용하는 방법 이외에는 없을 것 같아요!이미지 처리라면 img size 를 줄이거나 특정 부분 feature(많이 사용되는)만 학습시키는 것도 방법 일 것 같구요

예은 · Answer

batch size는 디폴트가 2 입니다 # scale sampling #conf.test_scale &#61; 512 #conf.crop_size &#61; [512, 1760] conf.test_scale &#61; 2 conf.crop_size &#61; [2, 2] 이밎 사이즈도 과하게 줄여봤지만 큰 효과는 보지 못했습니다ㅠ 특정 부분 feature(많이 사용되는)만 학습시키는 것은 어떻게 하는지 혹시 설명 가능 하신가요ㅠㅠ?

CUDA out of memory 문제(파이토치)

2개의 답변

#cuda 목록