3
0
Fork 0
forked from mirrors/nixpkgs

python3Packages.cupy: use the latest cudnn

This commit is contained in:
Samuel Ainsworth 2022-04-01 08:02:47 +00:00
parent e51d84a97f
commit 2a0acdbe9f

View file

@ -1961,7 +1961,7 @@ in {
cupy = callPackage ../development/python-modules/cupy {
cudatoolkit = pkgs.cudatoolkit_11;
cudnn = pkgs.cudnn_8_1_cudatoolkit_11;
cudnn = pkgs.cudnn_8_3_cudatoolkit_11;
nccl = pkgs.nccl_cudatoolkit_11;
cutensor = pkgs.cutensor_cudatoolkit_11;
};