diff options
| -rw-r--r-- | var/spack/repos/builtin/packages/py-torch/package.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py index 065deb08d2..75125cd08c 100644 --- a/var/spack/repos/builtin/packages/py-torch/package.py +++ b/var/spack/repos/builtin/packages/py-torch/package.py @@ -223,12 +223,14 @@ class PyTorch(PythonPackage, CudaPackage): @property def libs(self): - root = join_path(site_packages_dir, 'torch', 'lib') + root = join_path(self.prefix, self.spec['python'].package.site_packages_dir, + 'torch', 'lib') return find_libraries('libtorch', root) @property def headers(self): - root = join_path(site_packages_dir, 'torch', 'include') + root = join_path(self.prefix, self.spec['python'].package.site_packages_dir, + 'torch', 'include') headers = find_all_headers(root) headers.directories = [root] return headers |