diff options
| -rw-r--r-- | var/spack/repos/builtin/packages/cudnn/package.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/var/spack/repos/builtin/packages/cudnn/package.py b/var/spack/repos/builtin/packages/cudnn/package.py index 6c9f1fe933..3fe161fbc6 100644 --- a/var/spack/repos/builtin/packages/cudnn/package.py +++ b/var/spack/repos/builtin/packages/cudnn/package.py @@ -250,9 +250,11 @@ class Cudnn(Package): if pkg: version(long_ver, sha256=pkg) # Add constraints matching CUDA version to cuDNN version - cuda_req = "cuda@{0}.0:{0}".format(cuda_ver) - cudnn_ver_req = "@{0}".format(long_ver) - depends_on(cuda_req, when=cudnn_ver_req) + # cuDNN builds for CUDA 11.x are compatible with all CUDA 11.x: + # https://docs.nvidia.com/deeplearning/cudnn/support-matrix/index.html#fntarg_2 + if Version(cuda_ver) >= Version("11"): + cuda_ver = Version(cuda_ver).up_to(1) + depends_on("cuda@{}".format(cuda_ver), when="@{}".format(long_ver)) def url_for_version(self, version): # Get the system and machine arch for building the file path |