hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

cudnn: adjust version dependency on CUDA (#31763)

Browse source
This commit is contained in:
Melven Roehrig-Zoellner 2022-08-07 17:27:51 +02:00 committed by GitHub
parent 86be4666fc
commit 67154a1088
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
var/spack/repos/builtin/packages/cudnn/package.py
View file

@ -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