Skip to content

Commit

Permalink
Merge pull request #3473 from easybuilders/ocaisa-patch-1
Browse files Browse the repository at this point in the history
enhance cuDNN easyblock to verify that EULA is accepted before installing it
  • Loading branch information
boegel authored Oct 9, 2024
2 parents bd7fe7e + 2137f38 commit 11afb88
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions easybuild/easyblocks/c/cudnn.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,3 +68,13 @@ def __init__(self, *args, **kwargs):
self.cfg['keepsymlinks'] = True
self.cfg.template_values['cudnnarch'] = cudnnarch
self.cfg.generate_template_values()

def fetch_step(self, *args, **kwargs):
"""Check for EULA acceptance prior to getting sources."""
# EULA for cuDNN must be accepted via --accept-eula-for EasyBuild configuration option,
# or via 'accept_eula = True' in easyconfig file
self.check_accepted_eula(
name='cuDNN',
more_info='https://docs.nvidia.com/deeplearning/cudnn/latest/reference/eula.html'
)
return super(EB_cuDNN, self).fetch_step(*args, **kwargs)

0 comments on commit 11afb88

Please sign in to comment.