From 6de3786c36053f413595986e6a91b94e23b3e6b1 Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Sat, 8 Apr 2023 12:05:34 -0500 Subject: cuDNN: LD_LIBRARY_PATH required (#36466) --- var/spack/repos/builtin/packages/cudnn/package.py | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'var') diff --git a/var/spack/repos/builtin/packages/cudnn/package.py b/var/spack/repos/builtin/packages/cudnn/package.py index cfe5641489..29234bdfdc 100644 --- a/var/spack/repos/builtin/packages/cudnn/package.py +++ b/var/spack/repos/builtin/packages/cudnn/package.py @@ -307,9 +307,18 @@ class Cudnn(Package): 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) -- cgit v1.2.3-60-g2f50