https://flyte.org logo
#ask-the-community
Title
# ask-the-community
s

seunggs

03/24/2023, 11:36 PM
Hi, we can specify the number of gpus in flyte tasks, but how do we specify which type of gpu to use?
k

Ketan (kumare3)

03/25/2023, 2:04 AM
Currently you cannot directly have to use pod templates or pod spec
@task(task_config=Pod(pod_spec=…))
?
k

Kevin Su

03/25/2023, 3:37 AM
yup
s

seunggs

03/25/2023, 3:37 AM
Great - thank you!