C#检测本机是否安装CUDA

打印 上一主题 下一主题

主题 948|帖子 948|积分 2844

假如你需要在 C# 中检测是否安装了支持 CUDA 的 GPU,可以使用 ManagedCuda 库。下面是一个简单的方法来检查 CUDA 设备的存在。
1.安装 ManagedCuda 库


2.检测 CUDA GPU

  1. using System;
  2. using ManagedCuda;
  3. public static bool HasCudaGPU()
  4. {
  5.     try
  6.     {
  7.         // 获取 CUDA 设备数量
  8.         int deviceCount = CudaContext.GetDeviceCount();
  9.         if (deviceCount > 0)
  10.         {
  11.             Console.WriteLine($"CUDA GPU detected: {deviceCount} device(s)");
  12.             for (int i = 0; i < deviceCount; i++)
  13.             {
  14.                 // 获取设备属性
  15.                 var deviceProperties = CudaContext.GetDeviceInfo(i);
  16.                 Console.WriteLine($"Device {i}: {deviceProperties.DeviceName}");
  17.                 Console.WriteLine($"  Compute Capability: {deviceProperties.ComputeCapability.Major}.{deviceProperties.ComputeCapability.Minor}");
  18.                 Console.WriteLine($"  Total Memory: {deviceProperties.TotalGlobalMemory / 1024 / 1024} MB");
  19.             }
  20.             return true; // 检测到 CUDA GPU
  21.         }
  22.         else
  23.         {
  24.             Console.WriteLine("No CUDA-enabled GPU detected.");
  25.             return false;
  26.         }
  27.     }
  28.     catch (CudaException ex)
  29.     {
  30.         // 捕获特定的 CUDA 异常并处理
  31.         Console.WriteLine("CUDA error: " + ex.Message);
  32.     }
  33.     catch (Exception ex)
  34.     {
  35.         Console.WriteLine("Error detecting CUDA GPU: " + ex.Message);
  36.     }
  37.     return false;
  38. }
复制代码
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

x
回复

使用道具 举报

0 个回复

正序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

勿忘初心做自己

金牌会员
这个人很懒什么都没写!

标签云

快速回复 返回顶部 返回列表