RuntimeError: CUDA out of memory. Tried to allocate 300.00 MiB (GPU 0; 4.00 GiB total capacity; 2.69 GiB already allocated; 220.35 MiB free; 2.71 GiB reserved in total by PyTorch)
결론부터 말하자면 위와 같은 에러가 발생했다면, mini_batch 사이즈를 줄이거나 이미지를 리사이징 하는 방법 정도 외에는 방법이 없다. (혹은 그래픽카드를 업그레이드/추가 하는 방법이 있긴 하다.)
작업관리자 상에서 cuda를 확인해보면 공유 GPU메모리가 7.9GB가 놀고있는게 보인다.
저걸 사용하는 방법을 찾아보려 하였으나, 사실 공유 GPU메모리는 GPU에 실존하는 메모리가 아니라 가상 메모리라서 그냥은 쓸 수가 없다고 한다. 공유 GPU메모리는 안쓰고 있는게 정상이다.
'반치용 > 문제해결(trouble shooting)' 카테고리의 다른 글
| 이미지 사이즈 일괄 변경 및 자르기 (0) | 2020.04.02 |
|---|---|
| [파이토치]윈도우에서 pycocotools 사용하는 방법 (0) | 2020.03.29 |
| cnn 파라미터 수 계산 방법 및 꿀팁 (size mismatch 에러 관련) (0) | 2020.03.29 |
| 파이토치를 이용한 resnet 기반 대장 내시경 종양 여부 구분 시도 (0) | 2020.03.28 |
| 파이토치/토치비전 설치 관련 (2) | 2020.03.07 |
댓글