[Perf] Add NUM_WORKERS and CPUS_PER_WORKER env to the mnist workload - Githubissues

ray-project / kuberay

A toolkit to run Ray applications on Kubernetes

Apache License 2.0

974 stars 330 forks source link

[Perf] Add NUM_WORKERS and CPUS_PER_WORKER env to the mnist workload #2126

Closed rueian closed 1 month ago

rueian commented 1 month ago

Why are these changes needed?

Hi @kevin85421, @andrewsykim,

I have made num_workers and the CPU part of resources_per_worker configurable with envs.

https://github.com/ray-project/kuberay/pull/2116#discussion_r1589696298

Related issue number

Checks

[ ] I've made sure the tests are passing.
Testing Strategy
- [ ] Unit tests
- [ ] Manual tests
- [ ] This PR is not tested :(