forked from mirrors/nixpkgs
python3Packages.cupy: use the latest cudnn
This commit is contained in:
parent
e51d84a97f
commit
2a0acdbe9f
|
@ -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;
|
||||
};
|
||||
|
|
Loading…
Reference in a new issue