diff --git a/EESSI-install-software.sh b/EESSI-install-software.sh
index aa7a3528..5057c891 100755
--- a/EESSI-install-software.sh
+++ b/EESSI-install-software.sh
@@ -456,6 +456,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 50548205..32072c5c 100644
--- a/eb_hooks.py
+++ b/eb_hooks.py
@@ -1711,11 +1711,15 @@ def pre_test_hook_increase_max_failed_tests_arm_PyTorch(self, *args, **kwargs):
     See https://github.com/EESSI/software-layer/issues/461
     """
     cpu_target = get_eessi_envvar('EESSI_SOFTWARE_SUBDIR')
-    if self.name == 'PyTorch' and self.version == '2.1.2':
-        if get_cpu_architecture() == AARCH64:
-            self.cfg['max_failed_tests'] = 10
-        if cpu_target in [CPU_TARGET_CASCADELAKE, CPU_TARGET_ICELAKE, CPU_TARGET_SAPPHIRE_RAPIDS]:
-            self.cfg['max_failed_tests'] = 4
+    if self.name == 'PyTorch':
+        if self.version == '2.1.2':
+            if get_cpu_architecture() == AARCH64:
+                self.cfg['max_failed_tests'] = 10
+            if cpu_target in [CPU_TARGET_CASCADELAKE, CPU_TARGET_ICELAKE, CPU_TARGET_SAPPHIRE_RAPIDS]:
+                self.cfg['max_failed_tests'] = 4
+        if self.version == '2.9.1':
+            if cpu_target == CPU_TARGET_A64FX:
+                self.cfg['max_failed_tests'] = 41
 
 
 def pre_test_hook_ignore_failing_tests_OpenBabel_a64fx(self, *args, **kwargs):
@@ -2203,6 +2207,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),
     },