From 13dd41882a8e58c6d1c41742068a42af4c6aba1d Mon Sep 17 00:00:00 2001 From: Thomas Roeblitz Date: Mon, 28 Oct 2024 13:11:41 +0100 Subject: [PATCH] tweak libpaths by adding directory containing libnccl.so.2 --- easybuild/easyblocks/t/tensorflow.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/easybuild/easyblocks/t/tensorflow.py b/easybuild/easyblocks/t/tensorflow.py index c72aec1ea1..b45bc8a781 100644 --- a/easybuild/easyblocks/t/tensorflow.py +++ b/easybuild/easyblocks/t/tensorflow.py @@ -562,6 +562,13 @@ def configure_step(self): tensorrt_root = get_software_root('TensorRT') nccl_root = get_software_root('NCCL') + # add path to libnccl.so.2 directory provided by NCCL when both sysroot + # and RPATH are used (such as in EESSI) + if build_option('sysroot') and self.toolchain.use_rpath: + libpaths = self.system_libs_info[2] + libpaths.append(os.path.join(nccl_root, 'lib')) + self.system_libs_info[2] = libpaths + self._with_cuda = bool(cuda_root) config_env_vars = {