python37Packages.cupy: Add cutensor optional dep

2021-03-15 23:09:15 +00:00
parent a680b02816
commit 13b5995f06
2 changed files with 3 additions and 1 deletions
--- a/pkgs/top-level/python-packages.nix
+++ b/pkgs/top-level/python-packages.nix
@@ -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 { };