fix nvml if it has not nvmlDeviceGetCudaComputeCapability dll function

2019-02-05 18:14:46 +04:00
parent c3b6330284
commit 0313fd9ae6
1 changed files with 5 additions and 1 deletions
--- a/nnlib/pynvml.py
+++ b/nnlib/pynvml.py
@@ -1704,7 +1704,11 @@ def nvmlDeviceGetTopologyCommonAncestor(device1, device2):
 def nvmlDeviceGetCudaComputeCapability(device):
    c_major = c_int()
    c_minor = c_int()
-    fn = _nvmlGetFunctionPointer("nvmlDeviceGetCudaComputeCapability")
+    
+    try:
+        fn = _nvmlGetFunctionPointer("nvmlDeviceGetCudaComputeCapability")
+    except:
+        return 9, 9
    
    # get the count
    ret = fn(device, byref(c_major), byref(c_minor))