CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix nvml if it has not nvmlDeviceGetCudaComputeCapability dll function

Browse Source
This commit is contained in:
iperov
2019-02-05 18:14:46 +04:00
parent c3b6330284
commit 0313fd9ae6
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nnlib/pynvml.py
+5 -1
View File
@@ -1704,7 +1704,11 @@ def nvmlDeviceGetTopologyCommonAncestor(device1, device2):
def nvmlDeviceGetCudaComputeCapability(device): def nvmlDeviceGetCudaComputeCapability(device):
c_major = c_int() c_major = c_int()
c_minor = c_int() c_minor = c_int()
fn = _nvmlGetFunctionPointer("nvmlDeviceGetCudaComputeCapability")
try:
fn = _nvmlGetFunctionPointer("nvmlDeviceGetCudaComputeCapability")
except:
return 9, 9
# get the count # get the count
ret = fn(device, byref(c_major), byref(c_minor)) ret = fn(device, byref(c_major), byref(c_minor))