제가 포트홀 감지를 할려고 yolov5를 돌리려고 합니다. 근데 맨 처음 시작하는 첫 단계에서는 되더니 그 다음부터 이와 같은 오류로 인해 작동이 안되고 실패합니다. 이런 cuda랑 cudnn이랑 torch를 맞추면 될 거 같은데 저는 맞췄다고 생각하는데 아니였나봐요. 왜 그런 지 알려주시면 감사합니다.
참고로
CUDA 11.3
cuDNN 8.4.0
torch 1.13.1+cu116
torchaudio 0.13.1+cu116
torchvision 0.14.1
gpu nvidia GeForce RTX 3060 입니다.
해결 완료했습니다
한 번 눌러주시길 부탁드립니다!안녕하세요! 우선 CUDA 11.3과 cuDNN 8.4.0을 사용하신다면 torch version이 1.13.1이 아닌 1.9.0 이상이어야 합니다. 또한, torch 1.9.0 버전부터는 cuDNN 8.2.2 이상이 필요합니다.
그리고 torch 1.9.0 이상 버전에 대해서는 CUDA 11.2, cuDNN 8.2.0 이상이 필요합니다.
따라서 가장 최신 버전인 CUDA 11.4, cuDNN 8.2.4, 그리고 torch 1.9.0 이상 버전을 설치해보시는 것을 추천드립니다.
만약 이에도 문제가 발생한다면, 해당 오류를 더 자세히 파악하기 위해서는 error 메세지와 traceback 내용 등을 공유해주시면 더 자세한 답변을 드릴 수 있을 것 같습니다.
혹시 어떤 식으로 문제 해결했는지도 남겨주실수 있으신지요?
다른 분들이 해결법을 알게 되면 많은 도움이 될 것 같습니다!
