diff options
-rw-r--r-- | var/spack/repos/builtin/packages/py-torch/package.py | 10 |
1 files changed, 8 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 b717d82dba..62f2b0cbf5 100644 --- a/var/spack/repos/builtin/packages/py-torch/package.py +++ b/var/spack/repos/builtin/packages/py-torch/package.py @@ -206,12 +206,18 @@ class PyTorch(PythonPackage, CudaPackage): @property def libs(self): - root = join_path(python_platlib, 'torch', 'lib') + # TODO: why doesn't `python_platlib` work here? + root = join_path( + self.prefix, self.spec['python'].package.platlib, 'torch', 'lib' + ) return find_libraries('libtorch', root) @property def headers(self): - root = join_path(python_platlib, 'torch', 'include') + # TODO: why doesn't `python_platlib` work here? + root = join_path( + self.prefix, self.spec['python'].package.platlib, 'torch', 'include' + ) headers = find_all_headers(root) headers.directories = [root] return headers |