티스토리 뷰

pip install 'onnxruntime-gpu<1.20'

 

2024년 11월10일 기준 분명 1.20버전이랑 1.19 버전이랑 recommended cuda, cudnn 버전이 큰 차이가 없어 보이는데 이상하게 1.20 버전으로 설치하면 cuda를 제대로 인식못한다.

 

참고로 pytorch/pytorch:2.4.1-cuda12.4-cudnn9-devel 이 이미지 base로 하다가 겪은 문제다.

 

https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html

 

NVIDIA - CUDA

Instructions to execute ONNX Runtime applications with CUDA

onnxruntime.ai

댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
Total
Today
Yesterday
링크
«   2025/01   »
1 2 3 4
5 6 7 8 9 10 11
12 13 14 15 16 17 18
19 20 21 22 23 24 25
26 27 28 29 30 31
글 보관함