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

pythonPackages.pytorch: add libnuma dep on Linux

Browse Source 
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
1 changed files with 4 additions and 3 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

@ -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