zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

cuDNN: LD_LIBRARY_PATH required (#36466)

Browse Source
This commit is contained in:
Adam J. Stewart 2023-04-08 12:05:34 -05:00 committed by GitHub
parent 3869761216
commit 6de3786c36
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
var/spack/repos/builtin/packages/cudnn/package.py
View File

@ -307,9 +307,18 @@ def url_for_version(self, version):
return url.format(directory, cuda, sys_key, ver)
def setup_run_environment(self, env):
# Package is not compiled, and does not work unless LD_LIBRARY_PATH is set
env.prepend_path("LD_LIBRARY_PATH", self.prefix.lib)
if "target=ppc64le: platform=linux" in self.spec:
env.set("cuDNN_ROOT", os.path.join(self.prefix, "targets", "ppc64le-linux"))
def setup_dependent_build_environment(self, env, dependent_spec):
self.setup_run_environment(env)
def setup_dependent_run_environment(self, env, dependent_spec):
self.setup_run_environment(env)
def install(self, spec, prefix):
install_tree(".", prefix)