diff --git a/EESSI-install-software.sh b/EESSI-install-software.sh
index e4834545..ccd96f76 100755
--- a/EESSI-install-software.sh
+++ b/EESSI-install-software.sh
@@ -396,6 +396,8 @@ else
             # load EasyBuild module (will be installed if it's not available yet)
             source ${TOPDIR}/load_easybuild_module.sh ${eb_version}
 
+            export EASYBUILD_HOOKS=$TOPDIR/eb_hooks.py
+
             ${EB} --show-config
 
             echo_green "All set, let's start installing some software with EasyBuild v${eb_version} in ${EASYBUILD_INSTALLPATH}..."
diff --git a/eb_hooks.py b/eb_hooks.py
index 82673010..c7791bd6 100644
--- a/eb_hooks.py
+++ b/eb_hooks.py
@@ -2021,6 +2021,12 @@ def set_maximum(parallel, max_value):
         '*': (divide_by_factor, 2),
         CPU_TARGET_A64FX: (set_maximum, 1),
     },
+    'PyTorch': {
+        CPU_TARGET_AARCH64_GENERIC: (divide_by_factor, 2),
+        CPU_TARGET_A64FX: (set_maximum, 8),
+        CPU_TARGET_NEOVERSE_N1: (divide_by_factor, 2),
+        CPU_TARGET_NEOVERSE_V1: (divide_by_factor, 2),
+    },
     'QuantumESPRESSO': {
         CPU_TARGET_A64FX: (set_maximum, 6),
     },