diff options
Diffstat (limited to 'nixpkgs/pkgs/development/python-modules/pytorch/default.nix')
-rw-r--r-- | nixpkgs/pkgs/development/python-modules/pytorch/default.nix | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/nixpkgs/pkgs/development/python-modules/pytorch/default.nix b/nixpkgs/pkgs/development/python-modules/pytorch/default.nix index f14200b3542..92afa80852a 100644 --- a/nixpkgs/pkgs/development/python-modules/pytorch/default.nix +++ b/nixpkgs/pkgs/development/python-modules/pytorch/default.nix @@ -2,7 +2,7 @@ cudaSupport ? false, cudatoolkit ? null, cudnn ? null, fetchFromGitHub, lib, numpy, pyyaml, cffi, typing, cmake, hypothesis, numactl, linkFarm, symlinkJoin, - utillinux, which }: + utillinux, which, isPy3k }: assert cudnn == null || cudatoolkit != null; assert !cudaSupport || cudatoolkit != null; @@ -27,6 +27,7 @@ let in buildPythonPackage rec { version = "1.0.0"; pname = "pytorch"; + disabled = !isPy3k; src = fetchFromGitHub { owner = "pytorch"; @@ -44,6 +45,7 @@ in buildPythonPackage rec { }) ]; + dontUseCmakeConfigure = true; preConfigure = lib.optionalString cudaSupport '' export CC=${cudatoolkit.cc}/bin/gcc CXX=${cudatoolkit.cc}/bin/g++ '' + lib.optionalString (cudaSupport && cudnn != null) '' |