From 3f1b0daf11a4c73198b894f3d58203578df9cd6c Mon Sep 17 00:00:00 2001
From: Yannick DAYER <yannick.dayer@idiap.ch>
Date: Tue, 26 Mar 2024 11:13:44 +0100
Subject: [PATCH] fix(slurm): add the worker-port conf for our setup
---
src/bob/pipelines/config/distributed/slurm_cpu_default.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/src/bob/pipelines/config/distributed/slurm_cpu_default.py b/src/bob/pipelines/config/distributed/slurm_cpu_default.py
index 74105ca..be6aa28 100644
--- a/src/bob/pipelines/config/distributed/slurm_cpu_default.py
+++ b/src/bob/pipelines/config/distributed/slurm_cpu_default.py
@@ -70,8 +70,12 @@ cluster = SLURMCluster(
protocol="tcp://",
scheduler_options={
"protocol": "tcp://",
- "port": 8786, # Workers will connect to the scheduler on that port.
+ "port": 8786, # Workers will connect to the scheduler on that port
},
+ worker_extra_args=[
+ "--worker-port",
+ "60001:63000", # Workers will be reachable by the Client on those ports
+ ],
)
cluster.adapt(
--
GitLab