Skip to content

Commit 8c06920

Browse files
author
Richard Top
committed
fixed hard-coded path in eb_hooks.py
1 parent 998037d commit 8c06920

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

eb_hooks.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -284,15 +284,14 @@ def parse_hook_tensorflow_CUDA(ec, eprefix):
284284

285285
ec['pretestopts'] = (
286286
"""interppath=$(find "$EESSI_EPREFIX/lib64" -name 'ld-*' | grep -E 'so\\.1|so\\.2' | head -n1) && """
287-
"""patchelf --set-interpreter "$interppath" """
288-
""""%(builddir)s/%(name)s/bazel-root/0b9648e0837f9e5bb579e0e2e64adf3f/external/python_%(arch)s-unknown-linux-gnu/bin/python%(pyshortver)s" && """
287+
"""pybin=$(find "%(builddir)s/%(name)s/bazel-root/" -type f -path "*/external/python_%(arch)s-unknown-linux-gnu/bin/python%(pyshortver)s" | head -n1) && """
288+
"""patchelf --set-interpreter "$interppath" "$pybin" && """
289289
"""export LD_LIBRARY_PATH="$EBROOTCUDA/lib:$EBROOTCUDNN/lib:$EBROOTNCCL/lib:$LD_LIBRARY_PATH" && """
290290
)
291291

292292
ec['postinstallcmds'] = [
293293
'mkdir -p %(installdir)s/bin',
294294
'ln -s $EBROOTCUDA/bin/cuobjdump %(installdir)s/bin/cuobjdump',
295-
#'chmod 755 -R %(builddir)s',
296295
]
297296

298297
print_msg("TensorFlow-CUDA related changes have been applied")

0 commit comments

Comments
 (0)