search

FourierMourier / yolov8-onnx-cpp

YOLOv8 C++ Instance Segmentation, Pose Estimation and Object Detection with ONNX
Apache License 2.0
103 stars 23 forks source link

How to use CUDA #1

Closed YangSangWan closed 1 year ago

YangSangWan commented 1 year ago

I want use cuda. help me...

★ My environment

but /nn/onnx_model_base.cpp line 46

sessionOptions.AppendExecutionProvider_CUDA(cudaOption); shutdown program... not exist log... 😪

FourierMourier commented 1 year ago

Hi! I haven't added some additional code for GPU usage yet but I'll do it sometime later, so don't worry also using cuda requires having another package with "gpu" part in its name: onnxruntime-win-x64-gpu-1.16.1.zip

YangSangWan commented 1 year ago

oh i solved this problem!

  1. in this page "https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html" check out "Requirements" my Environment CUDA 11.8 ->onnxruntime 1.15

  2. download onnxruntime! this page "https://github.com/microsoft/onnxruntime/releases" onnxruntime-win-x64-gpu-1.15.0.zip

    thank you

FourierMourier commented 1 year ago

I'm glad it helped