fix

Signed-off-by: Hemil Desai <hemild@nvidia.com>

Author: hemildesai

nemo_run/run/ray/slurm.py

@@ -186,7 +186,7 @@ def get_gres_specification() -> str:
             if self.executor.gres:
                 return f"--gres={self.executor.gres}"
             elif self.executor.gpus_per_node:
-                return f"gpu:{self.executor.gpus_per_node}"
+                return f"--gres=gpu:{self.executor.gpus_per_node}"
             else:
                 return ""