pytorch-bin: init at 1.6.0

2020-08-30 11:12:16 -07:00
parent d214fce42c
commit 33b21aec60
4 changed files with 85 additions and 1 deletions
@@ -4100,6 +4100,10 @@ in {
    cudaSupport = pkgs.config.cudaSupport or false;
  };

+  pytorch-bin = callPackage ../development/python-modules/pytorch/bin.nix {
+    inherit (pkgs.linuxPackages) nvidia_x11;
+  };
+
  pyro-ppl = callPackage ../development/python-modules/pyro-ppl {};

  opt-einsum = if isPy27 then