勿忘初心做自己 发表于 2025-2-18 22:58:04

C#检测本机是否安装CUDA

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

https://i-blog.csdnimg.cn/direct/fafc5e0563194c9381935c3d5320bbbb.png
2.检测 CUDA GPU

using System;
using ManagedCuda;

public static bool HasCudaGPU()
{
    try
    {
      // 获取 CUDA 设备数量
      int deviceCount = CudaContext.GetDeviceCount();
      if (deviceCount > 0)
      {
            Console.WriteLine($"CUDA GPU detected: {deviceCount} device(s)");
            for (int i = 0; i < deviceCount; i++)
            {
                // 获取设备属性
                var deviceProperties = CudaContext.GetDeviceInfo(i);
                Console.WriteLine($"Device {i}: {deviceProperties.DeviceName}");
                Console.WriteLine($"Compute Capability: {deviceProperties.ComputeCapability.Major}.{deviceProperties.ComputeCapability.Minor}");
                Console.WriteLine($"Total Memory: {deviceProperties.TotalGlobalMemory / 1024 / 1024} MB");
            }
            return true; // 检测到 CUDA GPU
      }
      else
      {
            Console.WriteLine("No CUDA-enabled GPU detected.");
            return false;
      }
    }
    catch (CudaException ex)
    {
      // 捕获特定的 CUDA 异常并处理
      Console.WriteLine("CUDA error: " + ex.Message);
    }
    catch (Exception ex)
    {
      Console.WriteLine("Error detecting CUDA GPU: " + ex.Message);
    }
    return false;
}

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页: [1]
查看完整版本: C#检测本机是否安装CUDA