mirror of
https://github.com/NixOS/nixpkgs.git
synced 2025-01-21 20:34:06 +00:00
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 New Issue
Block a user