public NVIDIAGpuSet(INTMinerRoot root)
{
_root = root;
this.Properties = new List <GpuSetProperty>();
if (NvmlInit())
{
NvmlNativeMethods.nvmlDeviceGetCount(ref deviceCount);
for (int i = 0; i < deviceCount; i++)
{
Gpu gpu = Gpu.Create(i, string.Empty);
nvmlDevice nvmlDevice = new nvmlDevice();
var nvmlReturn = NvmlNativeMethods.nvmlDeviceGetHandleByIndex((uint)i, ref nvmlDevice);
SetGpuStatus(gpu, nvmlReturn);
NvmlNativeMethods.nvmlDeviceGetName(nvmlDevice, out string name);
nvmlMemory memory = new nvmlMemory();
NvmlNativeMethods.nvmlDeviceGetMemoryInfo(nvmlDevice, ref memory);
// short gpu name
if (!string.IsNullOrEmpty(name))
{
name = name.Replace("GeForce GTX ", string.Empty);
name = name.Replace("GeForce ", string.Empty);
}
gpu.Name = name;
gpu.TotalMemory = memory.total;
_gpus.Add(i, gpu);
}
if (deviceCount > 0)
{
NvmlNativeMethods.nvmlSystemGetDriverVersion(out _driverVersion);
NvmlNativeMethods.nvmlSystemGetNVMLVersion(out string nvmlVersion);
this.Properties.Add(new GpuSetProperty(GpuSetProperty.DRIVER_VERSION, "驱动版本", _driverVersion));
try {
double driverVersionNum;
if (double.TryParse(_driverVersion, out driverVersionNum))
{
var item = root.SysDicItemSet.GetSysDicItems("CudaVersion")
.Select(a => new { Version = double.Parse(a.Value), a })
.OrderByDescending(a => a.Version)
.FirstOrDefault(a => driverVersionNum >= a.Version);
if (item != null)
{
this.Properties.Add(new GpuSetProperty("CudaVersion", "Cuda版本", item.a.Code));
}
}
}
catch (Exception e) {
Logger.ErrorDebugLine(e);
}
this.Properties.Add(new GpuSetProperty("NVMLVersion", "NVML版本", nvmlVersion));
Dictionary <string, string> kvs = new Dictionary <string, string> {
{ "CUDA_DEVICE_ORDER", "PCI_BUS_ID" }
};
foreach (var kv in kvs)
{
var property = new GpuSetProperty(kv.Key, kv.Key, kv.Value);
this.Properties.Add(property);
}
Task.Factory.StartNew(() => {
foreach (var gpu in _gpus.Values)
{
NVIDIAOverClock.RefreshGpuState(gpu);
}
// 这里会耗时5秒
foreach (var kv in kvs)
{
Environment.SetEnvironmentVariable(kv.Key, kv.Value);
}
});
}
}
}
public NVIDIAGpuSet(INTMinerRoot root) : this()
{
_root = root;
if (Design.IsInDesignMode)
{
return;
}
string nvsmiDir = Path.Combine(Environment.GetFolderPath(System.Environment.SpecialFolder.ProgramFiles), "NVIDIA Corporation", "NVSMI");
if (Directory.Exists(nvsmiDir))
{
Windows.NativeMethods.SetDllDirectory(nvsmiDir);
NvmlNativeMethods.nvmlInit();
_isNvmlInited = true;
NvmlNativeMethods.nvmlDeviceGetCount(ref deviceCount);
for (int i = 0; i < deviceCount; i++)
{
nvmlDevice nvmlDevice = new nvmlDevice();
NvmlNativeMethods.nvmlDeviceGetHandleByIndex((uint)i, ref nvmlDevice);
uint gClock = 0, mClock = 0;
NvmlNativeMethods.nvmlDeviceGetName(nvmlDevice, out string name);
NvmlNativeMethods.nvmlDeviceGetMaxClockInfo(nvmlDevice, nvmlClockType.Graphics, ref gClock);
NvmlNativeMethods.nvmlDeviceGetMaxClockInfo(nvmlDevice, nvmlClockType.Mem, ref mClock);
if (!string.IsNullOrEmpty(name))
{
name = name.Replace("GeForce ", string.Empty);
}
Gpu gpu = new Gpu {
Index = i,
Name = name,
Temperature = 0,
PowerUsage = 0,
FanSpeed = 0,
OverClock = new NVIDIAOverClock()
};
_gpus.Add(i, gpu);
}
if (deviceCount > 0)
{
NvmlNativeMethods.nvmlSystemGetDriverVersion(out string driverVersion);
NvmlNativeMethods.nvmlSystemGetNVMLVersion(out string nvmlVersion);
this.Properties.Add(new GpuSetProperty("DriverVersion", "driver version", driverVersion));
this.Properties.Add(new GpuSetProperty("NVMLVersion", "NVML version", nvmlVersion));
Dictionary <string, string> kvs = new Dictionary <string, string> {
{ "CUDA_DEVICE_ORDER", "PCI_BUS_ID" }
};
foreach (var kv in kvs)
{
var property = new GpuSetProperty(kv.Key, kv.Key, kv.Value);
this.Properties.Add(property);
}
Task.Factory.StartNew(() => {
// 这里会耗时5秒
foreach (var kv in kvs)
{
Environment.SetEnvironmentVariable(kv.Key, kv.Value);
}
foreach (var gpu in _gpus.Values)
{
NVIDIAOverClock.RefreshGpuState(gpu);
}
});
}
}
}