mirror of
https://github.com/NixOS/nixpkgs.git
synced 2025-01-22 04:45:39 +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 {
|
cupy = callPackage ../development/python-modules/cupy {
|
||||||
cudatoolkit = pkgs.cudatoolkit_11;
|
cudatoolkit = pkgs.cudatoolkit_11;
|
||||||
cudnn = pkgs.cudnn_8_1_cudatoolkit_11;
|
cudnn = pkgs.cudnn_8_3_cudatoolkit_11;
|
||||||
nccl = pkgs.nccl_cudatoolkit_11;
|
nccl = pkgs.nccl_cudatoolkit_11;
|
||||||
cutensor = pkgs.cutensor_cudatoolkit_11;
|
cutensor = pkgs.cutensor_cudatoolkit_11;
|
||||||
};
|
};
|
||||||
|
Loading…
Reference in New Issue
Block a user