···

       
25
       
25
       
 
       
  installPhase = ''

     

       
26
       
26
       
 
       
    function fixRunPath {

     

       
27
       
27
       
 
       
      p=$(patchelf --print-rpath $1)

     

       
28
       
28
       
-
       
      patchelf --set-rpath "$p:${lib.makeLibraryPath [ stdenv.cc.cc ]}" $1

     

       
28
       
28
       
+
       
      patchelf --set-rpath "''${p:+$p:}${lib.makeLibraryPath [ stdenv.cc.cc ]}:\$ORIGIN/" $1

     

       
29
       
29
       
 
       
    }

     

       
30
       
30
       
-
       
    fixRunPath lib64/libcudnn.so

     

       
30
       
30
       
+
       


     

       
31
       
31
       
+
       
    for lib in lib64/lib*.so; do

     

       
32
       
32
       
+
       
      fixRunPath $lib

     

       
33
       
33
       
+
       
    done

     

       
31
       
34
       
 
       


     

       
32
       
35
       
 
       
    mkdir -p $out

     

       
33
       
36
       
 
       
    cp -a include $out/include

     
···

       
38
       
41
       
 
       
  # See the explanation in addOpenGLRunpath.

     

       
39
       
42
       
 
       
  postFixup = ''

     

       
40
       
43
       
 
       
    for lib in $out/lib/lib*.so; do

     

       
41
       
41
       
-
       
      # patchelf fails on libcudnn_cnn_infer due to it being too big.

     

       
42
       
42
       
-
       
      # Most programs will still get the RPATH since they link to

     

       
43
       
43
       
-
       
      # other things.

     

       
44
       
44
       
-
       
      # (https://github.com/NixOS/patchelf/issues/222)

     

       
45
       
45
       
-
       
      if [ "$(basename $lib)" != libcudnn_cnn_infer.so ]; then

     

       
46
       
46
       
-
       
        addOpenGLRunpath $lib

     

       
47
       
47
       
-
       
      fi

     

       
44
       
44
       
+
       
      addOpenGLRunpath $lib

     

       
48
       
45
       
 
       
    done

     

       
49
       
46
       
 
       
  '';

     

       
50
       
47