Commit a43d296d authored by David Guibert's avatar David Guibert
Browse files

cudnn: add 8.4.0

parent 79bc66ce

pkgs/development/libraries/science/math/cudnn/extension.nix

+17 −1
Original line number Diff line number Diff line
@@ -97,6 +97,22 @@ final: prev: let 
        supportedCudaVersions = [ "11.0" "11.1" "11.2" "11.3" "11.4" "11.4" "11.5" "11.6" "11.7" ];

      }

    ];

    "8.4.0" = [

      rec {

        fileVersion = "10.2";

        fullVersion = "8.4.0.27";

        hash = "sha256-FMXjykJYJxmW0f2VnELRfFgs5Nmv9FH4RSRGnnhP0VQ=";

        url = "${urlPrefix}/v${majorMinorPatch fullVersion}/local_installers/${fileVersion}/cudnn-linux-x86_64-${fullVersion}_cuda${fileVersion}-archive.tar.xz";

        supportedCudaVersions = [ "10.2" ];

      }

      rec {

        fileVersion = "11.6";

        fullVersion = "8.4.0.27";

        hash = "sha256-0Zva/ZgAx50p5vb/+p+eLBDREy1sL/ELFZPgV+dN0FA=";

        url = "${urlPrefix}/v${majorMinorPatch fullVersion}/local_installers/${fileVersion}/cudnn-linux-x86_64-${fullVersion}_cuda${fileVersion}-archive.tar.xz";

        supportedCudaVersions = [ "11.0" "11.1" "11.2" "11.3" "11.4" "11.4" "11.5" "11.6" "11.7" ];

      }

    ];

  };



  # Default attributes
@@ -111,7 +127,7 @@ final: prev: let 
    "11.4" = "8.3.2";

    "11.5" = "8.3.2";

    "11.6" = "8.3.2";

    "11.7" = "8.3.2";

    "11.7" = "8.4.0";

  }.${cudaVersion};



in cuDnnPackages