티스토리 뷰
Deep Learning
정말 내가 분명 gpu driver랑 cuda, cudnn 문제 없이 설치하고 설정 잘 해줬는데 onnxruntime-gpu 라이브러리에서 gpu인식이 제대로 안된다면
developer0hye 2024. 11. 10. 20:38pip 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
'Deep Learning' 카테고리의 다른 글
huggingface transformers cache directory 변경(웨이트 저장되는 폴더 변경) 방법 (0) | 2024.11.19 |
---|---|
배경 제거 모델 RMBG-2.0 (2) | 2024.11.18 |
LLM 과 Few shot 관련된 논문 (3) | 2024.11.07 |
huggingface transformers pipeline gpu 지정해서 모델 load하는 방법 (0) | 2024.11.06 |
GOT OCR 2.0은 한국어 OCR이 가능한가? (0) | 2024.10.26 |
댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- 백준 11437
- Lowest Common Ancestor
- C++ Deploy
- MOT
- 인공지능을 위한 선형대수
- FairMOT
- 백준
- 백준 11053
- 문제집
- 백트래킹
- 백준 1766
- 자료구조
- PyCharm
- cosine
- ㅂ
- 순열
- 가장 긴 증가하는 부분 수열
- 위상 정렬 알고리즘
- 이분탐색
- 조합
- 파이참
- LCA
- 단축키
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
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 |
글 보관함