Можно ли назначить задачу конкретному работнику в Ray?

Question

В частности, я бы хотел, чтобы мой рабочий хранилища параметров всегда вызывался на узле HEAD, а не на каком-либо из рабочих. Таким образом, я могу оптимизировать конфигурацию ресурса. В настоящее время задача сохранения параметров запускается на случайном сервере, даже если он вызывается первым, и даже если за ним следует ray.get ()

Может быть, можно сделать что-то вроде: ps = ParameterStore.remote(onHead=True)

Answer 1

Вы можете запустить «головной» узел с дополнительным настраиваемым ресурсом, а затем вы можете заставить актера хранилища параметров потребовать этот настраиваемый ресурс.Например, начните головной узел с:

ray start --head --resources='{"PSResource": 1}'

Затем вы можете объявить класс субъекта хранилища параметров с помощью

@ray.remote(resources={"PSResource": 1})
class ParameterStore(object):
    pass


ps = ParameterStore.remote()

Вы также можете регулярно объявлять субъект хранилища параметров и менять способВы вызываете это.Например,

@ray.remote
class ParameterStore(object):
    pass


ps = ParameterStore._remote(args=[], resources={"PSResource": 1})

Подробнее о ресурсах в Ray можно узнать по адресу https://ray.readthedocs.io/en/latest/resources.html.