about summary refs log tree commit diff
path: root/pkgs/development/python-modules/tensorflow
diff options
context:
space:
mode:
authorLuke Granger-Brown <git@lukegb.com>2021-04-24 01:10:47 +0000
committerLuke Granger-Brown <git@lukegb.com>2021-04-24 01:12:19 +0000
commitd385b8f5972d4a0885c29c6bac87471bf622c133 (patch)
tree8b3e8c70896584fc076b6960786b066d6245ce9e /pkgs/development/python-modules/tensorflow
parentc298ef8855f6ad629da5e09aab36ee673473316c (diff)
python3Packages.tensorflow: mark as requiring big-parallel
The Bazel build part of tensorflow builds can really use as many cores
as is available - to avoid blocking a small machine for literally hours
while it churns away, mark the build as big-parallel so we can schedule
it to run quickly on a machine with lots of cores.
Diffstat (limited to 'pkgs/development/python-modules/tensorflow')
-rw-r--r--pkgs/development/python-modules/tensorflow/default.nix4
1 files changed, 4 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/tensorflow/default.nix b/pkgs/development/python-modules/tensorflow/default.nix
index 0e2df1a4678da..0216c05eeacf1 100644
--- a/pkgs/development/python-modules/tensorflow/default.nix
+++ b/pkgs/development/python-modules/tensorflow/default.nix
@@ -325,6 +325,10 @@ let
           addOpenGLRunpath "$lib"
         done
       '';
+
+      requiredSystemFeatures = [
+        "big-parallel"
+      ];
     };
 
     meta = with lib; {