python 환경에서 onnxruntime-gpu를 설치하여 사용하려는데
자꾸만 "ImportError: cannot import name 'get_all_providers'" 가 발생했다.
구글링 해봐도 같은 증상을 겪는 사람이 많은데 한참을 찾다 발견한 내용.
onnxruntime release note에서 내가 못본건지 명확히 안쓰여있었던건지 모르겠지만
cuda 버전에 따라 지원하는 onnxruntime 버전이 다르기 떄문에 맞춰서 사용해야했다.
다음번 참고를 위해 onnx runtime ver - cuda ver 정보를 기록
- onnx runtime v1.5.1 ~ v1.6.0 --> cuda 10.2 ( cuda 11 build from source)
- onnx runtime v1.2.0 ~ v1.4.0 --> cuda 10.1
- onnx runtime v0.5.0 ~ v1.1.2 --> cuda 10.0
- onnx runtime v0.2.1 ~ v0.4.0 --> cuda 9.1 (nuget packages --> cuda 10.0)
'전공관련 > Deep Learning' 카테고리의 다른 글
| [Pytorch] pycharm 환경에서 torch.distributed.launch를 실행하자 (1) | 2021.07.13 |
|---|---|
| [Tensorflow] h5py 관련 오류를 해결하자 (0) | 2021.04.26 |
| [용어] Ablation Study (0) | 2021.02.25 |
| [Caffe] caffe 환경 설정없이 caffemodel 값을 확인하자 (0) | 2021.02.24 |
| [Pytorch] model load시 key가 있는 레이어만 불러오자 (0) | 2021.02.17 |
매직블럭
작은 지식들 그리고 기억 한조각