From 60176e005dd466727bd9bde8fbeb01f39ddee37e Mon Sep 17 00:00:00 2001 From: Austin Seipp Date: Mon, 24 Dec 2018 19:59:40 -0600 Subject: [PATCH] pythonPackages.pytorch: add libnuma dep on Linux Signed-off-by: Austin Seipp --- pkgs/development/python-modules/pytorch/default.nix | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/pkgs/development/python-modules/pytorch/default.nix b/pkgs/development/python-modules/pytorch/default.nix index 11fe0b5e6c4..fd3ada70762 100644 --- a/pkgs/development/python-modules/pytorch/default.nix +++ b/pkgs/development/python-modules/pytorch/default.nix @@ -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