about summary refs log tree commit diff
diff options
context:
space:
mode:
authorSamuel Ainsworth <skainsworth@gmail.com>2023-07-26 11:42:05 -0700
committerGitHub <noreply@github.com>2023-07-26 11:42:05 -0700
commit3cfc4cff26375af2f8f5db73a6699d1da5dc6dfe (patch)
treedadf3591c0727c636d0655aa7ee640938de3ef12
parent78df3591ec67310b8cc4b753e1496999da2678cf (diff)
parent9325a80a11ed2bffa693d36e79112072769acdf4 (diff)
Merge pull request #243729 from ogoid/fixup-missing-cudnn
cudaPackages.cudnn: fixes libcublas not found error #236070
-rw-r--r--pkgs/development/libraries/science/math/cudnn/generic.nix1
1 files changed, 1 insertions, 0 deletions
diff --git a/pkgs/development/libraries/science/math/cudnn/generic.nix b/pkgs/development/libraries/science/math/cudnn/generic.nix
index cdfa924b2242a..e0a6ffd9d547b 100644
--- a/pkgs/development/libraries/science/math/cudnn/generic.nix
+++ b/pkgs/development/libraries/science/math/cudnn/generic.nix
@@ -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 = {