pythonPackages.pytorch: add libnuma dep on Linux
Signed-off-by: Austin Seipp <aseipp@pobox.com>
This commit is contained in:
parent
79d21285dd
commit
60176e005d
@ -1,6 +1,6 @@
|
|||||||
{ fetchurl, buildPythonPackage, pythonOlder,
|
{ stdenv, fetchurl, buildPythonPackage, pythonOlder,
|
||||||
cudaSupport ? false, cudatoolkit ? null, cudnn ? null,
|
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,
|
linkFarm, symlinkJoin,
|
||||||
utillinux, which }:
|
utillinux, which }:
|
||||||
|
|
||||||
@ -77,7 +77,8 @@ in buildPythonPackage rec {
|
|||||||
numpy.blas
|
numpy.blas
|
||||||
utillinux
|
utillinux
|
||||||
which
|
which
|
||||||
] ++ lib.optionals cudaSupport [cudatoolkit_joined cudnn];
|
] ++ lib.optionals cudaSupport [ cudatoolkit_joined cudnn ]
|
||||||
|
++ lib.optionals stdenv.isLinux [ numactl ];
|
||||||
|
|
||||||
propagatedBuildInputs = [
|
propagatedBuildInputs = [
|
||||||
cffi
|
cffi
|
||||||
|
Loading…
Reference in New Issue
Block a user