python3Packages.jaxlib: --set-rpath
-> --add-rpath
This commit is contained in:
parent
6e0570c14d
commit
1e239b3ae2
1 changed files with 1 additions and 1 deletions
|
@ -416,7 +416,7 @@ buildPythonPackage {
|
|||
ln -s ${cudaPackagesGoogle.cuda_nvcc.bin}/bin/ptxas $out/bin/ptxas
|
||||
|
||||
find $out -type f \( -name '*.so' -or -name '*.so.*' \) | while read lib; do
|
||||
patchelf --set-rpath "${cuda_libs_joined}/lib:${cudnn}/lib:${nccl}/lib:$(patchelf --print-rpath "$lib")" "$lib"
|
||||
patchelf --add-rpath "${lib.makeLibraryPath [cuda_libs_joined cudnn nccl]}" "$lib"
|
||||
done
|
||||
'';
|
||||
|
||||
|
|
Loading…
Reference in a new issue