nixpkgs/pkgs/development/libraries/science/math/cudnn/generic.nix
Daniël de Kok 8dad19aebe cudnn_cudatookit_11: add $ORIGIN/ to library RPATHs
Starting with version 8, CuDNN consists of multiple libraries.
Libraries (except libcudnn.so) are dlopen()ed on demand.

Since the CuDNN library path itself was not added to the rpath of the
libraries, use of CuDNN often fails with the following error:

Could not load library libcudnn_cnn_train.so.8.
Error: libcudnn_ops_train.so.8: cannot open shared object file: No such file or directory
Please make sure libcudnn_cnn_train.so.8 is in your library path!

This change fixes this by adding $ORIGIN/ to the rpath of all CuDNN
libraries.
2021-02-14 11:35:58 +01:00

65 lines
1.4 KiB
Nix

{ version
, srcName
, sha256
}:
{ stdenv
, lib
, cudatoolkit
, fetchurl
, addOpenGLRunpath
}:
stdenv.mkDerivation {
name = "cudatoolkit-${cudatoolkit.majorVersion}-cudnn-${version}";
inherit version;
src = fetchurl {
# URL from NVIDIA docker containers: https://gitlab.com/nvidia/cuda/blob/centos7/7.0/runtime/cudnn4/Dockerfile
url = "https://developer.download.nvidia.com/compute/redist/cudnn/v${version}/${srcName}";
inherit sha256;
};
nativeBuildInputs = [ addOpenGLRunpath ];
installPhase = ''
function fixRunPath {
p=$(patchelf --print-rpath $1)
patchelf --set-rpath "''${p:+$p:}${lib.makeLibraryPath [ stdenv.cc.cc ]}:\$ORIGIN/" $1
}
for lib in lib64/lib*.so; do
fixRunPath $lib
done
mkdir -p $out
cp -a include $out/include
cp -a lib64 $out/lib64
'';
# Set RUNPATH so that libcuda in /run/opengl-driver(-32)/lib can be found.
# See the explanation in addOpenGLRunpath.
postFixup = ''
for lib in $out/lib/lib*.so; do
addOpenGLRunpath $lib
done
'';
propagatedBuildInputs = [
cudatoolkit
];
passthru = {
inherit cudatoolkit;
majorVersion = lib.versions.major version;
};
meta = with lib; {
description = "NVIDIA CUDA Deep Neural Network library (cuDNN)";
homepage = "https://developer.nvidia.com/cudnn";
license = licenses.unfree;
platforms = [ "x86_64-linux" ];
maintainers = with maintainers; [ mdaiter ];
};
}