pythonPackages.pytorch: add libnuma dep on Linux

Signed-off-by: Austin Seipp <aseipp@pobox.com>
This commit is contained in:
Austin Seipp 2018-12-24 19:59:40 -06:00
parent 79d21285dd
commit 60176e005d

View file

@ -1,6 +1,6 @@
{ fetchurl, buildPythonPackage, pythonOlder,
{ stdenv, fetchurl, buildPythonPackage, pythonOlder,
cudaSupport ? false, cudatoolkit ? null, cudnn ? null,
fetchFromGitHub, lib, numpy, pyyaml, cffi, typing, cmake, hypothesis,
fetchFromGitHub, lib, numpy, pyyaml, cffi, typing, cmake, hypothesis, numactl,
linkFarm, symlinkJoin,
utillinux, which }:
@ -77,7 +77,8 @@ in buildPythonPackage rec {
numpy.blas
utillinux
which
] ++ lib.optionals cudaSupport [cudatoolkit_joined cudnn];
] ++ lib.optionals cudaSupport [ cudatoolkit_joined cudnn ]
++ lib.optionals stdenv.isLinux [ numactl ];
propagatedBuildInputs = [
cffi