指定特定设备类型

目录

指定特定设备类型

有时任务可能希望运行在某种特定类型的 GPU 上,可以在 Pod 注解中填写 nvidia.com/use-gputype 字段。HAMi 调度器会检查 nvidia-smi -L 返回的设备类型是否包含注解中的内容。

例如,带有以下注解的任务将被分配到 A100 或 V100 GPU:

metadata:
  annotations:
    nvidia.com/use-gputype: "A100,V100" # 为该作业指定显卡类型,使用逗号分隔,不会在非指定显卡上启动作业

任务也可以使用 nvidia.com/nouse-gputype 来避开某些类型的 GPU。在下面的示例中,该作业不会被分配到 1080(包括 1080Ti)或 2080(包括 2080Ti)类型的显卡。

metadata:
  annotations:
    nvidia.com/nouse-gputype: "1080,2080" # 为该作业指定黑名单显卡类型,使用逗号分隔,不会在指定显卡上启动作业