pytorch: Use nativeBuildInputs to specify binary build dependencies.

2019-03-06 21:34:26 +00:00 · 2019-03-06 21:34:26 +00:00 · 488230c253
parent 80d032b13b
commit 488230c253
1 changed files with 5 additions and 2 deletions
--- a/pkgs/development/python-modules/pytorch/default.nix
+++ b/pkgs/development/python-modules/pytorch/default.nix
@ -77,11 +77,14 @@ in buildPythonPackage rec {
  # https://github.com/intel/mkl-dnn/commit/8134d346cdb7fe1695a2aa55771071d455fae0bc
  NIX_CFLAGS_COMPILE = lib.optionals (numpy.blasImplementation == "mkl") [ "-Wno-error=array-bounds" ];

-  buildInputs = [
+  nativeBuildInputs = [
     cmake
-     numpy.blas
     utillinux
     which
+  ];
+
+  buildInputs = [
+     numpy.blas
  ] ++ lib.optionals cudaSupport [ cudatoolkit_joined cudnn ]
    ++ lib.optionals stdenv.isLinux [ numactl ];