Merge pull request ilkarman#65 from vivgupt/patch-1

Update to utils.py for Macs having CUDA installed

Thanks Vivek!!

Author: miguelgfierro

notebooks/common/utils.py

@@ -36,16 +36,14 @@ def get_cuda_version():
     """Get CUDA version"""
     if sys.platform == 'win32':
         raise NotImplementedError("Implement this!")
-    elif sys.platform == 'linux':
+    elif sys.platform == 'linux' or sys.platform == 'darwin':
         path = '/usr/local/cuda/version.txt'
         if os.path.isfile(path):
             with open(path, 'r') as f:
                 data = f.read().replace('\n','')
             return data
         else:
             return "No CUDA in this machine"
-    elif sys.platform == 'darwin':
-        raise NotImplementedError("Find a Mac with GPU and implement this!")
     else:
         raise ValueError("Not in Windows, Linux or Mac")
 
@@ -78,7 +76,27 @@ def get_cudnn_version():
         else:
             return "No CUDNN in this machine"
     elif sys.platform == 'darwin':
-        raise NotImplementedError("Find a Mac with GPU and implement this!")
+        candidates = ['/usr/local/cuda/include/cudnn.h',
+                      '/usr/include/cudnn.h']
+        for c in candidates:
+            file = glob.glob(c)
+            if file: break
+        if file:
+            with open(file[0], 'r') as f:
+                version = ''
+                for line in f:
+                    if "#define CUDNN_MAJOR" in line:
+                        version = line.split()[-1]
+                    if "#define CUDNN_MINOR" in line:
+                        version += '.' + line.split()[-1]
+                    if "#define CUDNN_PATCHLEVEL" in line:
+                        version += '.' + line.split()[-1]
+            if version:
+                return version
+            else:
+                return "Cannot find CUDNN version"
+        else:
+            return "No CUDNN in this machine"
     else:
         raise ValueError("Not in Windows, Linux or Mac")
 
@@ -351,4 +369,4 @@ def get_train_valid_test_split(n, train=0.7, valid=0.1, test=0.2, shuffle=False)
                                            test_size=test/other_split,
                                            shuffle=False)
     print("train:{} valid:{} test:{}".format(len(train_set), len(valid_set), len(test_set)))
-    return train_set, valid_set, test_set
+    return train_set, valid_set, test_set