使用Python提交Slurm作业

import subprocess

def submit_slurm_job(script_path, job_name, num_nodes, num_tasks_per_node, output_file):
    command = f"sbatch --job-name={job_name} --nodes={num_nodes} --ntasks-per-node={num_tasks_per_node} --output={output_file} {script_path}"
    subprocess.run(command, shell=True)

# 提交一个名为"my_job"的Slurm作业，使用2个节点，每个节点4个任务，输出结果到"output.txt"
submit_slurm_job("my_script.sh", "my_job", 2, 4, "output.txt")