[YOLO - darkflow] CUDA_ERROR_OUT_OF_MEMORY, CUDNN_STATUS_ALLOC_FAILED 오류 해결 방법

 

블로그 인기글

● 메이플스토리 신규 룬패치/자동해제 프로그램 [링크 이동]

● 메이플스토리 거짓말탐지기 알림 프로그램 [링크 이동]

● C#에서 TensorFlow 사용하는법 [링크 이동]

---

 

※ CUDA_ERROR_OUT_OF_MEMORY, CUDNN_STATUS_ALLOC_FAILED 오류 해결 방법

   →그래픽 카드 사양이...좋지 않기때문에...............생기는 오류같습니다...

     gpu 와 batch를 이렇게 수정하면 됩니다. ㅠㅅㅠ

 

    (dell 7577 inspiron 노트북 gtx 1060max-q 기준)

 

     gpu 1.0 = gpu를 100% 쓰겠다. 이므로 0.3 정도로 하면 가까스로 돌아갑니다.

     0.4부터 제 노트북은 오류가 납니다..

 

python flow --model cfg/yolo.cfg --load bin/yolo.weights --demo videofile.mp4 --saveVideo ↓ python flow --demo videofile.mp4 --model cfg/yolo.cfg --saveVideo --load -1 --gpu 0.3 batch 5

 

커스텀 학습 시에도 

python flow --model cfg/yolo.cfg --load bin/yolo.weights --train --annoation train/Annotations --dataset train/Images --load -1 --gpu 0.3 --batch 5  형태로 넣어주세요.