Skip to content

Commit 8347c65

Browse files
author
deshevoy
committed
[build] Change default cuDNN version to 8.6.0
To match CUDA12 To use the previous version add `-DCUDNN_VERSION=8.0.5` build flag commit_hash:c4d0b8ac7640a1c6dad5c2993723ff632a99c29b
1 parent 7000a56 commit 8347c65

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

build/ymake_conf.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2522,10 +2522,10 @@ def __init__(self, cuda):
25222522
self.cudnn_version = Setting('CUDNN_VERSION', auto=self.auto_cudnn_version)
25232523

25242524
def have_cudnn(self):
2525-
return self.cudnn_version.value in ('7.6.5', '8.0.5')
2525+
return self.cudnn_version.value in ('7.6.5', '8.0.5', '8.6.0')
25262526

25272527
def auto_cudnn_version(self):
2528-
return '8.0.5'
2528+
return '8.6.0'
25292529

25302530
def print_(self):
25312531
if self.cuda.have_cuda.value and self.have_cudnn():

0 commit comments

Comments
 (0)