Does gpu parameter in `@resource` decorater works ...
# ask-metaflow
m
Does gpu parameter in
@resource
decorater works well? When using with
--with kubernetes:image=my_image
in cli run, the job in k8s requests cpu&memory as written in resource decorator but it doesn't request gpu. I wanna check if this problem happens only in my setting or not.
1
s
how have you configured the gpu machines in the kubernetes cluster?
which is the recommended way to configure GPUs on top of EKS/AKS/GKE
m
yes it works well if I write all resources in cli parameter. e.g.) --with kubernetes:image=my_image,gpu=1,cpu=2,memory=3
s
and it doesn't if you use
@kubernetes
?
m
With
@kubernetes
decorator, it also works well. With
@resource
decorator +
--with kubernetes
, no it doesn't/
s
that may be a regression with Metaflow
m
Yes I guess so if it's reproducible for others. I thought it is a problem in the point of cpu and memory are requested well with that combination.
s
I can see a todo in the code base noting this issue. We will get this addressed and make a release next Monday. Thanks for pointing this out!
😲 1
👍 1
m
Thanks for your quick response!
🙌 1
fixed at 2.7.15 thanks!