Does `@torch_run` support kubernetes? In the pos...
# ask-metaflow
c
Does
@torch_run
support kubernetes? In the post from last year it says not yet. https://outerbounds.com/blog/distributed-training-with-metaflow/ But I think I see support in the repo here https://github.com/outerbounds/metaflow-torchrun
✅ 1
a
yes, it does now. you would need to install jobsets on your kubernetes cluster and if you are running at scale, kueue as well.
c
What do you mean "scale" exactly,
a
to get access to true gang scheduling semantics, you would need both jobsets and kueue. just using jobsets can result in deadlocks - the probability is rare if you are not running many workloads and have ample compute capacity available.
it's been on our backlog to publish more details about this set up in open source
c
Interesting I definitely haven't done that yet so.