c#调用CUDA执行YOLOV5对象检测
  aQQ2uVWFfQnM 2023年11月02日 42 0


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快。。。

【版权声明】本文内容来自摩杜云社区用户原创、第三方投稿、转载,内容版权归原作者所有。本网站的目的在于传递更多信息,不拥有版权,亦不承担相应法律责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@moduyun.com

上一篇: CUDA11.6下载地址 下一篇: 医疗终端札记
  1. 分享:
最后一次编辑于 2023年11月08日 0

暂无评论

推荐阅读
  anLrwkgbyYZS   2023年12月30日   32   0   0 ideciciMaxideMax
aQQ2uVWFfQnM