onnxruntime
-
[onnxruntime] CUDAExecutionProvider가 적용되지 않을 때Programing/기타 오류 해결 2024. 5. 8. 20:21
GPU 딸린 새 VM에 환경을 구축하고 테스트를 진행하는데Applied privider가 'CPUExecutionProvider'로 적용되고 있습니다. CUDA, cuDNN, onnxruntime 간 버전이 맞지 않아 발생한 문제일 가능성이 큽니다. 클라우드 서비스에서 제공하는 GPU VM은 적어도 CUDA, cuDNN 간 버전은 맞춰진 상태이므로,onnxruntime만 조정해보도록 하겠습니다. CUDA, cuDNN 버전 확인CUDAnvcc -V cuDNNcat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2 ONNX Runtime 재설치이미 onnxruntime이 설치가 되어있었다면 충돌문제가 발생할 수 있기 때문에 먼저 삭제해주세..