diff --git a/pkgs/development/libraries/science/math/cudnn/extension.nix b/pkgs/development/libraries/science/math/cudnn/extension.nix index 8226f4bc7ee9..817ba070223e 100644 --- a/pkgs/development/libraries/science/math/cudnn/extension.nix +++ b/pkgs/development/libraries/science/math/cudnn/extension.nix @@ -28,7 +28,9 @@ final: prev: let # Add all supported builds as attributes allBuilds = mapAttrs' (version: file: nameValuePair (computeName version) (buildCuDnnPackage (removeAttrs file ["fileVersion"]))) supportedVersions; # Set the default attributes, e.g. cudnn = cudnn_8_3_1; - defaultBuild = { "cudnn" = allBuilds.${computeName cuDnnDefaultVersion}; }; + defaultBuild = { "cudnn" = if allBuilds ? ${computeName cuDnnDefaultVersion} + then allBuilds.${computeName cuDnnDefaultVersion} + else throw "cudnn-${cuDnnDefaultVersion} does not support your cuda version ${cudaVersion}"; }; in allBuilds // defaultBuild; cuDnnVersions = let