Hugh Q. Lee | AI, 로보틱스, 개발 블로그

  • 홈
  • 태그

CUDA 1

[PyTorch/YOLO 오류 해결] GPU 미사용 및 torchvision::nms CUDA 에러 해결 방법

YOLO 모델을 사용할 때 GPU가 동작하지 않거나, 다음과 같은 에러가 발생할 수 있습니다.Could not run 'torchvision::nms' with arguments from the 'CUDA' backend.이 문제를 해결한 과정을 정리합니다.0. 문제 요약YOLO 모델을 사용하는데 GPU가 사용되지 않음.torchvision::nms 연산에서 CUDA 백엔드 오류 발생.특히 torchvision::nms 오류는 torch와 torchvision의 빌드가 다른 경우에 발생합니다:나의 경우에는, torch는 CUDA 빌드(예: +cu118)인데, torchvision은 CPU 전용 빌드(예: +cpu)일 때, 발생함.1. CUDA 버전 확인터미널에 입력:nvcc --version나의 경우 출..

개발 2025.04.29
이전
1
다음
더보기
프로필사진

Hugh Q. Lee | AI, 로보틱스, 개발 블로그

AI, 컴퓨터 비전, 로보틱스 프로젝트를 개발하고 기록하는 Hugh Q. Lee의 기술 블로그입니다.

  • 분류 전체보기 (51) N
    • 생각 (9)
    • 여행 (6)
    • 개발 (36) N
      • AI (4)
      • Bio (1)
      • Jetson (4)
      • Raspberry Pi (4) N
      • Robotics (1)

Tag

세이노의 가르침, raspi, Deep Learning, Image Classification, CV, 노개북, python, 라즈베리파이4, DeepLearning, 사이드프로젝트, 해몽해드림, object detection, pyqt5, DART, 경주여행, jetson, openai, Computer Vision, 노마드코더, 북클럽,

최근글과 인기글

  • 최근글
  • 인기글

방문자수Total

  • Today :
  • Yesterday :
250x250
반응형
홈페이지

Copyright © Kakao Corp. All rights reserved.

  • 홈페이지

티스토리툴바