diff --git a/pkgs/top-level/all-packages.nix b/pkgs/top-level/all-packages.nix index 0fb17d0ef7c85..60e1b9bf93c9a 100644 --- a/pkgs/top-level/all-packages.nix +++ b/pkgs/top-level/all-packages.nix @@ -4615,7 +4615,7 @@ with pkgs; cudatoolkit_11_5 cudatoolkit_11_6; - cudatoolkit = cudatoolkit_10; + cudatoolkit = cudatoolkit_11; cudnnPackages = callPackages ../development/libraries/science/math/cudnn { }; inherit (cudnnPackages) @@ -4638,8 +4638,7 @@ with pkgs; cudnn_8_3_cudatoolkit_10 cudnn_8_3_cudatoolkit_11; - # TODO(samuela): This is old and should be upgraded to 8.3 at some point. - cudnn = cudnn_7_6_cudatoolkit_10_1; + cudnn = cudnn_8_3_cudatoolkit_11; cutensorPackages = callPackages ../development/libraries/science/math/cutensor { }; inherit (cutensorPackages)