Merge pull request #81 from gwtnb/cudnn760

Add cuDNN 7.6.0
unnonouno · Jul 2, 2019 · 3bf48ea · 3bf48ea
2 parents a25760f + ab999da
commit 3bf48ea
Show file tree

Hide file tree

Showing 2 changed files with 54 additions and 0 deletions.
diff --git a/cudnnenv/__init__.py b/cudnnenv/__init__.py
@@ -349,6 +349,30 @@
         sha256sum='2c833f43c9147d9a25a20947a4c5a5f5c33b2443240fd767f63b330c482e68e0',
     )
 
+    codes['v7.6.0-cuda9'] = cudnn_base.format(
+        cudnn='cudnn-9.0-linux-x64-v7.6.0.64',
+        cudnn_ver='v7.6.0',
+        sha256sum='90659ea77734b7b671afe930c9898d21a13b888998f1dd3940cc57d6b2f29b86',
+    )
+
+    codes['v7.6.0-cuda92'] = cudnn_base.format(
+        cudnn='cudnn-9.2-linux-x64-v7.6.0.64',
+        cudnn_ver='v7.6.0',
+        sha256sum='ff028e6f07349445c16fef704a90bccb0992c3e012bba66ab1da352bad55b304',
+    )
+
+    codes['v7.6.0-cuda10'] = cudnn_base.format(
+        cudnn='cudnn-10.0-linux-x64-v7.6.0.64',
+        cudnn_ver='v7.6.0',
+        sha256sum='c4e1ee4168f4cadabaa989487a47bed09f34d34e35398b6084a2699d11bd2560',
+    )
+
+    codes['v7.6.0-cuda101'] = cudnn_base.format(
+        cudnn='cudnn-10.1-linux-x64-v7.6.0.64',
+        cudnn_ver='v7.6.0',
+        sha256sum='e956c6f9222fcb867a10449cfc76dee5cfd7c7531021d95fe9586d7e043b57d7',
+    )
+
     codes['v71-cuda8'] = codes['v7.1.3-cuda8']
     codes['v71-cuda9'] = codes['v7.1.4-cuda9']
     codes['v71-cuda91'] = codes['v7.1.3-cuda91']
@@ -365,6 +389,10 @@
     codes['v75-cuda92'] = codes['v7.5.1-cuda92']
     codes['v75-cuda10'] = codes['v7.5.1-cuda10']
     codes['v75-cuda101'] = codes['v7.5.1-cuda101']
+    codes['v76-cuda9'] = codes['v7.6.0-cuda9']
+    codes['v76-cuda92'] = codes['v7.6.0-cuda92']
+    codes['v76-cuda10'] = codes['v7.6.0-cuda10']
+    codes['v76-cuda101'] = codes['v7.6.0-cuda101']
 
     LIBDIR = 'lib64'
 
@@ -515,13 +543,27 @@
         sha256sum='2035071b11ffd7da8473f47a63ffd6610cb73108e072ff44b708ec833c218b45',
     )
 
+    codes['v7.6.0-cuda10'] = cudnn_base.format(
+        cudnn='cudnn-10.0-osx-x64-v7.6.0.64',
+        cudnn_ver='v7.6.0',
+        sha256sum='30ea0ef517d1f8a2be29a13bdefa9abd71d952d0cddfb8b2c3fba36a48c46f0f',
+    )
+
+    codes['v7.6.0-cuda101'] = cudnn_base.format(
+        cudnn='cudnn-10.1-osx-x64-v7.6.0.64',
+        cudnn_ver='v7.6.0',
+        sha256sum='8d1417e8f528cfda4e859cf41ecb843bad0a8920935ce0dadde5f7327d8427fb',
+    )
+
     codes['v7-cuda9'] = codes['v7.0.3-cuda9']
     codes['v71-cuda92'] = codes['v7.1.4-cuda92']
     codes['v72-cuda92'] = codes['v7.2.1-cuda92']
     codes['v73-cuda10'] = codes['v7.3.0-cuda10']
     codes['v74-cuda10'] = codes['v7.4.2-cuda10']
     codes['v75-cuda10'] = codes['v7.5.1-cuda10']
     codes['v75-cuda101'] = codes['v7.5.1-cuda101']
+    codes['v76-cuda10'] = codes['v7.6.0-cuda10']
+    codes['v76-cuda101'] = codes['v7.6.0-cuda101']
 
     LIBDIR = 'lib'
 

diff --git a/test/test_command.py b/test/test_command.py
@@ -71,6 +71,10 @@
   v7.5.1-cuda101
   v7.5.1-cuda9
   v7.5.1-cuda92
+  v7.6.0-cuda10
+  v7.6.0-cuda101
+  v7.6.0-cuda9
+  v7.6.0-cuda92
   v71-cuda8
   v71-cuda9
   v71-cuda91
@@ -87,6 +91,10 @@
   v75-cuda101
   v75-cuda9
   v75-cuda92
+  v76-cuda10
+  v76-cuda101
+  v76-cuda9
+  v76-cuda92
 '''
 elif sys.platform == 'darwin':
     _available_versions = '''  v2
@@ -111,12 +119,16 @@
   v7.5.0-cuda101
   v7.5.1-cuda10
   v7.5.1-cuda101
+  v7.6.0-cuda10
+  v7.6.0-cuda101
   v71-cuda92
   v72-cuda92
   v73-cuda10
   v74-cuda10
   v75-cuda10
   v75-cuda101
+  v76-cuda10
+  v76-cuda101
 '''