diff options
Diffstat (limited to 'pkgs/development/python-modules/pytorch/default.nix')
| -rw-r--r-- | pkgs/development/python-modules/pytorch/default.nix | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/pytorch/default.nix b/pkgs/development/python-modules/pytorch/default.nix index 7c45659220cb..f14200b3542b 100644 --- a/pkgs/development/python-modules/pytorch/default.nix +++ b/pkgs/development/python-modules/pytorch/default.nix @@ -81,11 +81,11 @@ in buildPythonPackage rec { cmake utillinux which - ]; + ] ++ lib.optionals cudaSupport [ cudatoolkit_joined ]; buildInputs = [ numpy.blas - ] ++ lib.optionals cudaSupport [ cudatoolkit_joined cudnn ] + ] ++ lib.optionals cudaSupport [ cudnn ] ++ lib.optionals stdenv.isLinux [ numactl ]; propagatedBuildInputs = [ |