public/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

pytorch: Use nativeBuildInputs to specify binary build dependencies.

Browse Source
This commit is contained in:
Tom Hunger 2019-03-06 21:34:26 +00:00 committed by Frederik Rietdijk
parent 80d032b13b
commit 488230c253
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pkgs/development/python-modules/pytorch/default.nix
View File

@ -77,11 +77,14 @@ in buildPythonPackage rec {
# https://github.com/intel/mkl-dnn/commit/8134d346cdb7fe1695a2aa55771071d455fae0bc # https://github.com/intel/mkl-dnn/commit/8134d346cdb7fe1695a2aa55771071d455fae0bc
NIX_CFLAGS_COMPILE = lib.optionals (numpy.blasImplementation == "mkl") [ "-Wno-error=array-bounds" ]; NIX_CFLAGS_COMPILE = lib.optionals (numpy.blasImplementation == "mkl") [ "-Wno-error=array-bounds" ];
buildInputs = [ nativeBuildInputs = [
cmake cmake
numpy.blas
utillinux utillinux
which which
];
buildInputs = [
numpy.blas
] ++ lib.optionals cudaSupport [ cudatoolkit_joined cudnn ] ] ++ lib.optionals cudaSupport [ cudatoolkit_joined cudnn ]
++ lib.optionals stdenv.isLinux [ numactl ]; ++ lib.optionals stdenv.isLinux [ numactl ];