From 02648a6cc1816b7f0156a5a2b9dcee2c0fbd37cc Mon Sep 17 00:00:00 2001 From: Jonathan Ringer Date: Sat, 28 Sep 2019 18:49:37 -0700 Subject: [PATCH] python3Packages.pytorch: fix build --- pkgs/development/python-modules/pytorch/default.nix | 1 + 1 file changed, 1 insertion(+) diff --git a/pkgs/development/python-modules/pytorch/default.nix b/pkgs/development/python-modules/pytorch/default.nix index f14200b3542..17bf6f5cdab 100644 --- a/pkgs/development/python-modules/pytorch/default.nix +++ b/pkgs/development/python-modules/pytorch/default.nix @@ -44,6 +44,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) ''