如何制定作业在哪个机器上跑

1请先搜索,在提问哦,搜索的过程也是提炼明确问题的过程,如果没有找到相关的问题和答案,建议按照以下模板清晰的描述问题!
2问题模板
【Ray使用环境】生产
【Ray版本和类库】
Ray: 2.0.0
【问题描述】
我们公司在ray集群上部署了两种不一样的worker节点,一部分是带有GPU的机器,一部分是只有CPU的机器,但是目前我们发现很多只跑CPU的作业会跑在GPU上面,导致GPU的作业无法运行。请问一下是否有办法指定作业只跑在C PU的机器上面呢,比如用tag标记机器,然后作业只能跑在某些tag上面

目前ray还不支持node tag,但可以通过custom resources间接实现,可以参考 https://docs.ray.io/en/latest/ray-core/scheduling/resources.html#custom-resources

1 个赞