Add Sam's pip torch path

EricGustin · Jan 28, 2022 · 52f13b6 · 52f13b6
1 parent 63306b7
commit 52f13b6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/smartsim/_core/_install/builder.py b/smartsim/_core/_install/builder.py
@@ -351,7 +351,9 @@ def _move_torch_libs(self):
 
         # retrieve torch shared libraries and copy to the
         # smartsim/_core/lib/backends/redisai_torch/lib dir
-        pip_torch_path = Path(site.getsitepackages()[0]) / "torch"
+        # self.torch_dir should be /path/to/torch/share/cmake/Torch
+        # so we take the great grandparent here
+        pip_torch_path = Path(self.torch_dir).parent.parent.parent
         pip_torch_lib_path = pip_torch_path / "lib"
 
         self.copy_dir(pip_torch_lib_path, ss_rai_torch_lib_path, set_exe=True)