Hi, we can specify the number of gpus in flyte tasks, but how do we specify which type of gpu to use?

Currently you cannot directly have to use pod templates or pod spec

Like this you mean? https://docs.flyte.org/projects/cookbook/en/latest/auto/integrations/kubernetes/pod/pod.html#multiple-containers

yup

Great - thank you!