python37Packages.cupy: Add cutensor optional dep

This commit is contained in:
John Ericson
2021-03-17 20:15:51 +00:00
parent a680b02816
commit 13b5995f06
2 changed files with 3 additions and 1 deletions
+1
View File
@@ -1589,6 +1589,7 @@ in {
cudatoolkit = pkgs.cudatoolkit_10_0;
cudnn = pkgs.cudnn_cudatoolkit_10_0;
nccl = pkgs.nccl_cudatoolkit_10;
cutensor = pkgs.cutensor_cudatoolkit_10;
};
curio = callPackage ../development/python-modules/curio { };