From 077a1cff14165d06e65db8de9027dcc7d3804526 Mon Sep 17 00:00:00 2001 From: Thomas MK Date: Tue, 3 Oct 2023 09:49:10 +0000 Subject: [PATCH] Add config for sharded kyiv --- external_confs/hydra/launcher/slurm/kyiv_shard.yaml | 12 ++++++++++++ 1 file changed, 12 insertions(+) create mode 100644 external_confs/hydra/launcher/slurm/kyiv_shard.yaml diff --git a/external_confs/hydra/launcher/slurm/kyiv_shard.yaml b/external_confs/hydra/launcher/slurm/kyiv_shard.yaml new file mode 100644 index 00000000..b35bbfc3 --- /dev/null +++ b/external_confs/hydra/launcher/slurm/kyiv_shard.yaml @@ -0,0 +1,12 @@ +--- +# usage: hydra/launcher=slurm/kyiv_shard + +defaults: + - submitit_slurm + +partition: kyiv +cpus_per_task: 12 # on kyiv, we have 24 CPUs per GPU +timeout_min: 99999 # 99999 minutes = a few months + +additional_parameters: + gres: "shard:5" # 5 shards à 8GB