Error: cuDNN isn't found FWD algo for convolution. 해결 방법

 

YOLOv4를 윈도우에서 컴파일 했다.

마스크를 착용한 사람/마스크를 착용하지 않은 사람을 구분하는 예제를 Train하려는데

Error: cuDNN isn't found FWD algo for convolution. 와 같은 에러가 발생했다.

 

여기저기 구글링을 해봤는데 의외로 해결 방법이 잘 안나와서 포스팅한다.

 

문제의 원인은 Input Data의 크기가 GPU Memory에 비해 너무 커서 발생하는 것이다.

 

해결하려면 더 좋은 GPU를 사용하거나, cfg 파일에서 Parameter 값을 조절해주면 된다.

지금 예제를 돌리는 노트북은 메모리가 2gb고, 다음과 같이 설정했을 때 무난하게 돌아간다.

 

 

트레이닝도 무난하게 되는 편이다. 

워낙 성능이 안좋은 GPU라 그런지 Train 시간이 10시간이 잡혔다.