Skip to content

Commit

Permalink
[build] Change default cuDNN version to 8.6.0
Browse files Browse the repository at this point in the history
To match CUDA12

To use the previous version add `-DCUDNN_VERSION=8.0.5` build flag
commit_hash:c4d0b8ac7640a1c6dad5c2993723ff632a99c29b
  • Loading branch information
deshevoy committed Feb 19, 2025
1 parent 08d1b9d commit 04378ca
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions build/ymake_conf.py
Original file line number Diff line number Diff line change
Expand Up @@ -2522,10 +2522,10 @@ def __init__(self, cuda):
self.cudnn_version = Setting('CUDNN_VERSION', auto=self.auto_cudnn_version)

def have_cudnn(self):
return self.cudnn_version.value in ('7.6.5', '8.0.5')
return self.cudnn_version.value in ('7.6.5', '8.0.5', '8.6.0')

def auto_cudnn_version(self):
return '8.0.5'
return '8.6.0'

def print_(self):
if self.cuda.have_cuda.value and self.have_cudnn():
Expand Down

0 comments on commit 04378ca

Please sign in to comment.