onnx docker 사용하기

특정  onnx runtime을 사용할때 도커 환경이 필요하다.
이때 도커를 사용하는것이 좋은것 같다.
 

아래 사이트에서 
https://github.com/microsoft/onnxruntime/tree/main/dockerfiles#cuda
 

이렇게 치면 도커가 설치됨.
sudo docker build -t onnxruntime-cuda -f ./dockerfiles/Dockerfile.cuda .
  
도커 명령어 모음. http://jmlim.github.io/docker/2019/02/24/docker-command/
 
버전 확인 https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html

NVIDIA - CUDA

ort 버전 https://onnxruntime.ai/docs/performance/model-optimizations/ort-format-models.html

ORT model format

 

https://github.com/aliyasineser/stereoDepth/blob/master/stereo_depth.py