diff options
author | ogoid <1449961+ogoid@users.noreply.github.com> | 2023-07-20 00:57:10 +0000 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-07-20 00:57:10 +0000 |
commit | e1c91b230122dceca76e600aae5a97d2f590ebb2 (patch) | |
tree | d23682cf0651efdce718f4968fa46d861a315072 /pkgs/development/libraries/science/math/cudnn | |
parent | c519ceaa21fd2fb14557a7543e4884cd6ead8b94 (diff) |
improve cudnn fix
Co-authored-by: Samuel Ainsworth <skainsworth@gmail.com>
Diffstat (limited to 'pkgs/development/libraries/science/math/cudnn')
-rw-r--r-- | pkgs/development/libraries/science/math/cudnn/generic.nix | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/pkgs/development/libraries/science/math/cudnn/generic.nix b/pkgs/development/libraries/science/math/cudnn/generic.nix index 99df51a59b208..dc9690cfc14e0 100644 --- a/pkgs/development/libraries/science/math/cudnn/generic.nix +++ b/pkgs/development/libraries/science/math/cudnn/generic.nix @@ -94,8 +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/*.so --add-needed libcublas.so.11 - patchelf $out/lib/*.so --add-needed libcublasLt.so.11 + patchelf $out/lib/libcudnn_ops_infer.so --add-needed libcublas.so.11 --add-needed libcublasLt.so.11 ''; passthru = { |