Merge pull request #3488 from boegel/20241015214234_new_pr_pytorch

also add path to `libtorch.so` & co to `$LIBRARY_PATH` in generated module file for PyTorch
easybuilders · Oct 16, 2024 · 5718bdf · 5718bdf
2 parents 3a6cb1a + 0fd1802
commit 5718bdf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/easybuild/easyblocks/p/pytorch.py b/easybuild/easyblocks/p/pytorch.py
@@ -571,6 +571,8 @@ def make_module_req_guess(self):
         guesses['CMAKE_PREFIX_PATH'] = [os.path.join(self.pylibdir, 'torch')]
         # Required to dynamically load libcaffe2_nvrtc.so
         guesses['LD_LIBRARY_PATH'] = [os.path.join(self.pylibdir, 'torch', 'lib')]
+        # important when RPATH linking is enabled
+        guesses['LIBRARY_PATH'] = [os.path.join(self.pylibdir, 'torch', 'lib')]
         return guesses