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

sticky-angle-28419

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?

freezing-airport-6809

03/25/2023, 2:04 AM

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

sticky-angle-28419

03/25/2023, 2:05 AM

sticky-angle-28419

03/25/2023, 2:06 AM

@task(task_config=Pod(pod_spec=…))

glamorous-carpet-83516

03/25/2023, 3:37 AM

yup

👍 1

sticky-angle-28419

03/25/2023, 3:37 AM

Great - thank you!

151 Views

Flyte

Flyte enables production-grade orchestration for machine learning workflows and data processing created to accelerate local workflows to production.