Mirrors/flowframes
1
0
Fork 0
mirror of https://github.com/n00mkrad/flowframes.git synced 2025-12-24 04:09:29 +01:00
Code Issues Packages Projects Releases Wiki Activity

RIFE-CUDA fp16 option and warplayer.py optimization

Browse Source
This commit is contained in:
N00MKRAD
2021-02-21 11:26:18 +01:00
parent 4a673744a5
commit 1d04d2d3ee
7 changed files with 79 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
Code/OS/NvApi.cs
View File

@@ -24,7 +24,7 @@ namespace Flowframes.OS
return;
gpu = gpus[0];
Logger.Log($"Initialized NvApi. GPU: {gpu.FullName}");
Logger.Log($"Initialized NvApi. GPU: {gpu.FullName} - Tensor Cores: {HasTensorCores()}");
}
catch (Exception e)
{
@@ -50,7 +50,7 @@ namespace Flowframes.OS
{
return (gpu.MemoryInformation.CurrentAvailableDedicatedVideoMemoryInkB / 1000f / 1024f);
}
catch (Exception e)
catch
{
return 0f;
}
@@ -72,5 +72,18 @@ namespace Flowframes.OS
return "";
}
}
public static bool HasTensorCores ()
{
if (gpu == null)
Init();
if (gpu == null)
return false;
string gpuName = gpu.FullName;
return (gpuName.Contains("RTX 20") || gpuName.Contains("RTX 30") || gpuName.Contains("Tesla V") || gpuName.Contains("Tesla T"));
}
}
}