cuda and cudnn version

Author: miguelgfierro

.gitignore

@@ -4,4 +4,5 @@
 .ipynb_checkpoints/
 cifar-10-batches-py/
 __pycache__
+.DS_Store
 

common/utils.py

@@ -26,6 +26,58 @@ def get_gpu_name():
         print(e)
 
 
+def get_cuda_version():
+    """Get CUDA version"""
+    if sys.platform == 'win32':
+        raise NotImplementedError("Implement this!")
+    elif sys.platform == 'linux':
+        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")
+
+
+def get_cudnn_version():
+    """Get CUDNN version"""
+    if sys.platform == 'win32':
+        raise NotImplementedError("Implement this!")
+    elif sys.platform == 'linux':
+        candidates = ['/usr/include/x86_64-linux-gnu/cudnn_v[0-99].h',
+                      '/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"
+    elif sys.platform == 'darwin':
+        raise NotImplementedError("Find a Mac with GPU and implement this!")
+    else:
+        raise ValueError("Not in Windows, Linux or Mac")
+
+
+
 def read_batch(src):
     '''Unpack the pickle files
     '''