Merge pull request #243729 from ogoid/fixup-missing-cudnn

cudaPackages.cudnn: fixes libcublas not found error #236070
2023-07-26 11:42:05 -07:00
parent 78df3591ec 9325a80a11
commit 3cfc4cff26
1 changed files with 1 additions and 0 deletions
@@ -94,6 +94,7 @@ in
    # Without --add-needed autoPatchelf forgets $ORIGIN on cuda>=8.0.5.
    postFixup = strings.optionalString (strings.versionAtLeast versionTriple "8.0.5") ''
      patchelf $out/lib/libcudnn.so --add-needed libcudnn_cnn_infer.so
+      patchelf $out/lib/libcudnn_ops_infer.so --add-needed libcublas.so --add-needed libcublasLt.so
    '';

    passthru = {