Skip to content

Commit 2252a75

Browse files
committed
Support version 7.5.0
1 parent cdc2790 commit 2252a75

File tree

2 files changed

+54
-0
lines changed

2 files changed

+54
-0
lines changed

cudnnenv/__init__.py

+42
Original file line numberDiff line numberDiff line change
@@ -301,6 +301,30 @@
301301
sha256sum='2edfc86a02b50d17e88c478955a332e6a1e8174e7e53a3458b4ea51faf02daa3',
302302
)
303303

304+
codes['v7.5.0-cuda9'] = cudnn_base.format(
305+
cudnn='cudnn-9.0-linux-x64-v7.5.0.56',
306+
cudnn_ver='v7.5.0',
307+
sha256sum='ee0ecd3cc30b9bf5ec875eac3ed375d3996bcb0ed5d2551716e4884b3ea5ce8c',
308+
)
309+
310+
codes['v7.5.0-cuda92'] = cudnn_base.format(
311+
cudnn='cudnn-9.2-linux-x64-v7.5.0.56',
312+
cudnn_ver='v7.5.0',
313+
sha256sum='2a04fd5ed5b8d32e2401c85a1a38f3cfd6da662c31bd26e80bea25469e48a675',
314+
)
315+
316+
codes['v7.5.0-cuda10'] = cudnn_base.format(
317+
cudnn='cudnn-10.0-linux-x64-v7.5.0.56',
318+
cudnn_ver='v7.5.0',
319+
sha256sum='701097882cb745d4683bb7ff6c33b8a35c7c81be31bac78f05bad130e7e0b781',
320+
)
321+
322+
codes['v7.5.0-cuda101'] = cudnn_base.format(
323+
cudnn='cudnn-10.1-linux-x64-v7.5.0.56',
324+
cudnn_ver='v7.5.0',
325+
sha256sum='c31697d6b71afe62838ad2e57da3c3c9419c4e9f5635d14b683ebe63f904fbc8',
326+
)
327+
304328
codes['v71-cuda8'] = codes['v7.1.3-cuda8']
305329
codes['v71-cuda9'] = codes['v7.1.4-cuda9']
306330
codes['v71-cuda91'] = codes['v7.1.3-cuda91']
@@ -313,6 +337,10 @@
313337
codes['v74-cuda9'] = codes['v7.4.2-cuda9']
314338
codes['v74-cuda92'] = codes['v7.4.2-cuda92']
315339
codes['v74-cuda10'] = codes['v7.4.2-cuda10']
340+
codes['v75-cuda9'] = codes['v7.5.0-cuda9']
341+
codes['v75-cuda92'] = codes['v7.5.0-cuda92']
342+
codes['v75-cuda10'] = codes['v7.5.0-cuda10']
343+
codes['v75-cuda101'] = codes['v7.5.0-cuda101']
316344

317345
LIBDIR = 'lib64'
318346

@@ -439,11 +467,25 @@
439467
sha256sum='4b1c53af7e50ff046ed53d4ba9129900da9bef0008b93cefad2e4195f1ee8ebb',
440468
)
441469

470+
codes['v7.5.0-cuda10'] = cudnn_base.format(
471+
cudnn='cudnn-10.0-osx-x64-v7.5.0.56',
472+
cudnn_ver='v7.5.0',
473+
sha256sum='5f72e5776eef6a4c5fc5c677404511faabc929864bd0886b884560e5bcdc8f54',
474+
)
475+
476+
codes['v7.5.0-cuda101'] = cudnn_base.format(
477+
cudnn='cudnn-10.1-osx-x64-v7.5.0.56',
478+
cudnn_ver='v7.5.0',
479+
sha256sum='58f7e5b51e3229cb294c1c134eef3ca1a0be1b3d12c3277d24d1912051f54840',
480+
)
481+
442482
codes['v7-cuda9'] = codes['v7.0.3-cuda9']
443483
codes['v71-cuda92'] = codes['v7.1.4-cuda92']
444484
codes['v72-cuda92'] = codes['v7.2.1-cuda92']
445485
codes['v73-cuda10'] = codes['v7.3.0-cuda10']
446486
codes['v74-cuda10'] = codes['v7.4.2-cuda10']
487+
codes['v75-cuda10'] = codes['v7.5.0-cuda10']
488+
codes['v75-cuda101'] = codes['v7.5.0-cuda101']
447489

448490
LIBDIR = 'lib'
449491

test/test_command.py

+12
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,10 @@
6363
v7.4.2-cuda10
6464
v7.4.2-cuda9
6565
v7.4.2-cuda92
66+
v7.5.0-cuda10
67+
v7.5.0-cuda101
68+
v7.5.0-cuda9
69+
v7.5.0-cuda92
6670
v71-cuda8
6771
v71-cuda9
6872
v71-cuda91
@@ -75,6 +79,10 @@
7579
v74-cuda10
7680
v74-cuda9
7781
v74-cuda92
82+
v75-cuda10
83+
v75-cuda101
84+
v75-cuda9
85+
v75-cuda92
7886
'''
7987
elif sys.platform == 'darwin':
8088
_available_versions = ''' v2
@@ -95,10 +103,14 @@
95103
v7.3.0-cuda10
96104
v7.4.1-cuda10
97105
v7.4.2-cuda10
106+
v7.5.0-cuda10
107+
v7.5.0-cuda101
98108
v71-cuda92
99109
v72-cuda92
100110
v73-cuda10
101111
v74-cuda10
112+
v75-cuda10
113+
v75-cuda101
102114
'''
103115

104116

0 commit comments

Comments
 (0)