MirrorHub/nixpkgs
0
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2024-11-17 15:22:59 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

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
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 ];