mirror of
https://github.com/NixOS/nixpkgs.git
synced 2026-10-30 16:10:36 +00:00
committed by
github-actions[bot]
parent
0f494fe08c
commit
0fe503acb2
@@ -5,7 +5,7 @@
|
||||
lib,
|
||||
libcublas,
|
||||
cuda_nvrtc,
|
||||
patchelf,
|
||||
buildPackages,
|
||||
zlib,
|
||||
}:
|
||||
buildRedist (
|
||||
@@ -39,8 +39,8 @@ buildRedist (
|
||||
# exist in CuDNN 8.
|
||||
# NOTE: Versions from CUDNN releases have four components.
|
||||
postFixup = lib.optionalString (cudnnAtLeast "8" && cudnnOlder "9") ''
|
||||
${lib.getExe patchelf} ''${!outputLib:?}/lib/libcudnn.so --add-needed libcudnn_cnn_infer.so
|
||||
${lib.getExe patchelf} ''${!outputLib:?}/lib/libcudnn_ops_infer.so --add-needed libcublas.so --add-needed libcublasLt.so
|
||||
${lib.getExe buildPackages.patchelf} ''${!outputLib:?}/lib/libcudnn.so --add-needed libcudnn_cnn_infer.so
|
||||
${lib.getExe buildPackages.patchelf} ''${!outputLib:?}/lib/libcudnn_ops_infer.so --add-needed libcublas.so --add-needed libcublasLt.so
|
||||
'';
|
||||
|
||||
# CuDNN depends on libnvrtc.so at runtime, as mentioned here in one small error description
|
||||
|
||||
Reference in New Issue
Block a user