728x90
1. Visual Studio Nuget 다운로드
2. VS include, lib path 설정
- includePath
- $(ProjectDir)/packages/Microsoft.ML.OnnxRuntime.Gpu.1.8.1/build/native/include
- LibPath
- $(ProjectDir)/packages/Microsoft.ML.OnnxRuntime.Gpu.1.8.1/runtimes/win-x64/native
- dll → 프로젝트 경로에 import
3. GPU 이용 시, 다음의 코드 변경
- core/framework/provider_options.h → provider_options.h
4. 테스트 코드 작성 후 테스트
728x90
'AI | ML > AI 개발 | CUDA' 카테고리의 다른 글
| [MetaFlow] ModuleNotFoundError: No module named 'fcntl' (0) | 2021.12.21 |
|---|---|
| [Python] TFLite to ONNX 변환 (0) | 2021.12.15 |
| [Onnx] C++ GPU Inference (2) | 2021.12.02 |
| [TensorRT] Windows 환경 onnx -> trt 변경 (0) | 2021.12.01 |
| [TensorRT] Windows 10 C++ 설치 (0) | 2021.12.01 |