YOLOv41 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시간이 잡혔다. 2021. 5. 18. 이전 1 다음
