File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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" )
You can’t perform that action at this time.
0 commit comments