gpu
argument in your function decorator.
T4
(16Gi)A10G
(24Gi)RTX4090
(24Gi)A100-40
(40Gi)H100
(80Gi)beam machine list
to check whether a machine is available.
gpu
parameter.
The list is ordered by priority. You can choose which GPUs to prioritize by specifying them at the front of the list.
T4
is prioritized over the A10G
, followed by the A100-40
.
gpu_count
parameter.