[ONNX] cuda 버전에 따른 python onnxruntime 버전을 맞추자.

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)