From 79d778f8cdb5f4f5eaaeb9a5d1215080698041e3 Mon Sep 17 00:00:00 2001 From: Robert Underwood Date: Tue, 10 Sep 2024 11:20:05 -0400 Subject: Add additional cuda-toolkit location variable used by py-torch (#46245) superceds #46128 Co-authored-by: Robert Underwood --- var/spack/repos/builtin/packages/py-torch/package.py | 1 + 1 file changed, 1 insertion(+) diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py index 9af10cb65a..2e1631fa02 100644 --- a/var/spack/repos/builtin/packages/py-torch/package.py +++ b/var/spack/repos/builtin/packages/py-torch/package.py @@ -526,6 +526,7 @@ class PyTorch(PythonPackage, CudaPackage, ROCmPackage): enable_or_disable("cuda") if "+cuda" in self.spec: + env.set("CUDA_TOOLKIT_ROOT_DIR", self.spec["cuda"].prefix) # Linux/macOS env.set("CUDA_HOME", self.spec["cuda"].prefix) # Linux/macOS env.set("CUDA_PATH", self.spec["cuda"].prefix) # Windows self.torch_cuda_arch_list(env) -- cgit v1.2.3-70-g09d2