반응형 torchvision::nms1 YOLOv8 torch 오류 해결 - Could not run ‘torchvision::nms’ with arguments from the ‘CUDA’ backend YOLOv8을 실행 중 다음과 같은 문제가 발생하였다. NotImplementedError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend NotImplementedError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend 오류는 PyTorch와 torchvision을 사용하여 객체 검출(Object Detection) 작업을 할 때 발생할 수 있다. 이 오류는 일반적으로 NVIDIA CUDA를 지원하는 GPU에서 발생하는데, 특히 torchvision의 nms (Non-Maximum Suppression) 함수를 호출할 때 CUDA 백엔드가.. 2024. 3. 31. 이전 1 다음 반응형
