#Assign to certain device type

#Assign to certain device type

Sometimes a task needs to run on a certain type of GPU. To do this, set the nvidia.com/use-gputype field in the pod annotations. The HAMi scheduler then checks whether the device type returned by nvidia-smi -L matches the annotation value.

For example, a task with the following annotation will be assigned to an A100 or V100 GPU:

metadata:
  annotations:
    nvidia.com/use-gputype: "A100,V100" # Specify the card type for this job, use comma to separate, will not launch job on non-specified card

A task can also use nvidia.com/nouse-gputype to avoid certain GPU types. In the following example, the job will not be assigned to 1080 (including 1080Ti) or 2080 (including 2080Ti) cards.

metadata:
  annotations:
    nvidia.com/nouse-gputype: "1080,2080" # Specify the blacklist card type for this job, use comma to separate, will not launch job on specified card