search

GoogleCloudPlatform / batch-samples

56 stars 28 forks source link

Add a simplified version of slurm to batch script supporting multinodes. #73

Closed zetxqx closed 4 months ago

zetxqx commented 4 months ago

The generated template is under slurm-to-batch/examples.

Tested by submitting a Batch job at slurm-to-batch/examples/exmample_batch_job.yaml. It is generated by replacing the placeholder in example_batch_job_template.yaml.

The job has two nodes and each node has 2 GPUs attached. The workload is printing the UUID of the GPU. Here is the result:

image

zetxqx commented 4 months ago

Removed the generated JSON batch template leaving only YAML format as well.