cudnn: init at 4.0

This commit is contained in:
Ian-Woo Kim 2016-05-24 10:14:25 +00:00
parent 5c68d2d1e6
commit e47070ad64
2 changed files with 40 additions and 0 deletions

View File

@ -0,0 +1,38 @@
{ stdenv, requireFile }:
stdenv.mkDerivation rec {
version = "4.0";
name = "cudnn-${version}";
src = requireFile rec {
name = "cudnn-7.0-linux-x64-v${version}-prod.tgz";
message = ''
This nix expression requires that ${name} is
already part of the store. Register yourself to NVIDIA Accelerated Computing Developer Program
and download cuDNN library at https://developer.nvidia.com/cudnn, and store it to the nix store with nix-store --add-fixed sha256 <FILE>.
'';
sha256 = "0zgr6qdbc29qw6sikhrh6diwwz7150rqc8a49f2qf37j2rvyyr2f";
};
phases = "unpackPhase installPhase fixupPhase";
installPhase = ''
mkdir -p $out
cp -a include $out/include
cp -a lib64 $out/lib64
'';
# all binaries are already stripped
#dontStrip = true;
# we did this in prefixup already
#dontPatchELF = true;
meta = {
description = "NVIDIA CUDA Deep Neural Network library (cuDNN)";
homepage = "https://developer.nvidia.com/cudnn";
license = stdenv.lib.licenses.unfree;
};
}

View File

@ -1201,6 +1201,8 @@ in
cudatoolkit = self.cudatoolkit7;
cudnn = callPackage ../development/libraries/science/math/cudnn/default.nix {};
curlFull = self.curl.override {
idnSupport = true;
ldapSupport = true;