From 0d4abe5d4b63d55f3b93cd744218cf519a9789a0 Mon Sep 17 00:00:00 2001 From: Luke Granger-Brown Date: Mon, 26 Apr 2021 00:44:48 +0000 Subject: [PATCH] python3Packages.pytorch: require big-parallel This compiles in usually about 2h15m with a 2-core build, but about 10m on a big-parallel machine. --- pkgs/development/python-modules/pytorch/default.nix | 3 +++ 1 file changed, 3 insertions(+) diff --git a/pkgs/development/python-modules/pytorch/default.nix b/pkgs/development/python-modules/pytorch/default.nix index 1436153e1db5..59a8c74f709c 100644 --- a/pkgs/development/python-modules/pytorch/default.nix +++ b/pkgs/development/python-modules/pytorch/default.nix @@ -297,6 +297,9 @@ in buildPythonPackage rec { install_name_tool -change @rpath/libc10.dylib $lib/lib/libc10.dylib $lib/lib/libshm.dylib ''; + # Builds in 2+h with 2 cores, and ~15m with a big-parallel builder. + requiredSystemFeatures = [ "big-parallel" ]; + meta = with lib; { description = "Open source, prototype-to-production deep learning platform"; homepage = "https://pytorch.org/";