c#调用CUDA执行YOLOV5对象检测

c#使用调YOLOV5对象检测,并调用CUDA进行计算

1.CUDA版本11.2

2.cuDNN用cudnn-windows-x86_64-8.9.3.28_cuda11-archive

记得把压缩包的三个文件夹放到cuda根目录下覆盖

3.Microsoft.ML.OnnxRuntime.Gpu要使用1.13.1,如果版本太新,SessionOptions会报错。

 private SessionOptions GetSessionOptions()
 {
     bool supportGPU = (Environment.GetEnvironmentVariable("CPAI_MODULE_SUPPORT_GPU") ?? "true").ToLower() == "true";

     SessionOptions sessionOpts = new SessionOptions();
     string[]? providers = null;
     try
     {
         providers = OrtEnv.Instance().GetAvailableProviders();
     }
     catch
     {
     }
     sessionOpts.AppendExecutionProvider_CUDA();//显卡用这个
     //sessionOpts.AppendExecutionProvider_CPU();
     return sessionOpts;
 }


//初始化onnx模型的代码

 SessionOptions sessionOpts = GetSessionOptions();

 _scorer = new YoloScorer<YoloCocoP5Model>("Assets/Weights/yolov5n.onnx", sessionOpts);

 _capture = new VideoCapture("http://192.168.151.130:8080/video"); // 手机摄像头流媒体

但是最终在2060笔记本显卡上发现 CUDA速度不如AMD的CPU快。。。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

xcagy

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值