From b8bb8a70cea7106a9eb71ad28815b00bdaf85edb Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Thu, 9 Nov 2023 16:25:54 -0600 Subject: PyTorch: specify CUDA root directory (#40855) --- var/spack/repos/builtin/packages/py-torch/package.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py index d2edd94538..782b0741d2 100644 --- a/var/spack/repos/builtin/packages/py-torch/package.py +++ b/var/spack/repos/builtin/packages/py-torch/package.py @@ -491,9 +491,8 @@ class PyTorch(PythonPackage, CudaPackage, ROCmPackage): enable_or_disable("cuda") if "+cuda" in self.spec: - # cmake/public/cuda.cmake - # cmake/Modules_CUDA_fix/upstream/FindCUDA.cmake - env.unset("CUDA_ROOT") + env.set("CUDA_HOME", self.spec["cuda"].prefix) # Linux/macOS + env.set("CUDA_PATH", self.spec["cuda"].prefix) # Windows torch_cuda_arch = ";".join( "{0:.1f}".format(float(i) / 10.0) for i in self.spec.variants["cuda_arch"].value ) -- cgit v1.2.3-60-g2f50