[OpenCV, Yolo, Cuda] 그래픽카드 변경으로 질문이 있어 글올립니다.

Question

안녕하세요 선배님들.현재 저희가 납품하는 영상처리 시스템이 있습니다.C&#43;&#43;로 Yolo v2, Cuda v10.0을 사용, 학습되어진 weights 파일을 활용하여 영상처리를 하는 방식입니다.(Yolo_v2 Library 사용)해당 S/W를 Installer로 패키징하여 배포하는 방식으로 납품하였습니다. 기존에 rtx 2070 GPU 서버에 설치하여 납품하였을때는 잘 동작하던 S/W가이번 그래픽카드 수급으로 인해 rtx 3060으로 변경하니 작동이 되지 않습니다. Yolo_v2_class.hpp에 있는 Detector를 선언하는 부분부터 소스 Flow가 멈추며 반응이 없으며 디버깅도 되지 않습니다.혹시 같은 문제를 겪으셨거나 해결 방안을 알고 계신 분 계시는지 도움받고자 글 올립니다. 감사합니다.

kimhwan · Accepted Answer

darknet 이용하신건가요? 저 같은 경우에는 그래픽카드에 따라서 정상적인 추론이 되지 않는 증상이 발생하여 찾아보니 빌드 전처리기에 CUDNN_HALF 옵션을 제거하니 정상적으로 동작하였습니다.자세히 찾아보진 않았지만 해당 옵션이 성능 최적화와 관련되어 더 빠르게 추론이나 학습이 가능하도록 하는 기능인 것 같은데 해당 기능이 그래픽카드에 따라서 ON 되고 OFF 되고 하는 것으로 확인하였습니다.

ITKim · Answer

&#64;kimhwan 감사합니다 확인해보겠습니다!

[OpenCV, Yolo, Cuda] 그래픽카드 변경으로 질문이 있어 글올립니다.

2개의 답변

#yolo 목록