vim /etc/crane/config.yaml

# 以上省略...
# 控制节点（主节点）
ControlMachine: crane01
# ...

# Nodes and partitions settings
# 计算节点
Nodes:
  - name: "crane[01-04]"
    #每个节点的资源
    cpu: 2
    memory: 2G
    #设备资源
    gres:
            #资源的name
          - name: gpu
            #资源的type
            type: a100
            # 设备资源对应的/dev设备文件，每个文件解析为一个设备
            DeviceFileRegex: /dev/nvidia[0-3]
            # 设备资源对应的/dev设备文件，/dev/dri/renderer[0-3]解析为一个设备
            DeviceFileList:
              - /dev/dri/renderer[0-3]
              - /dev/dri/renderer[4-7]
            #设备对应的环境变量 如CUDA_VISIABLE_DEVICES等
            #可选：nvidia,hip,ascend
            EnvInjector: nvidia

# partition information list
# 计算节点的分区
Partitions:
  - name: CPU              # 分区的名称（可自定义）
    nodes: "crane[01-02]"  # 分区中的节点，需要和 Nodes 部分对应
    priority: 5
  - name: GPU
    nodes: "crane[03-04]"
    priority: 3
    # Optional default memory per cpu in MB, 0 let scheduler decide
    DefaultMemPerCpu: 0     # 建议设置为0
    # Optional maximum memory per cpu in MB, 0 indicates no limit
    MaxMemPerCpu: 0         # 建议设置为0

# 默认分区，未指定分区的作业将被提交到默认分区，需要和 Partitions 部分对应
DefaultPartition: CPU