如何在光线平行和不使用光线模式的情况下运行函数？

Question

后 sudo pip3 install ray

我创建了一个函数foo()，在 ray 装饰器中定义：

import ray
ray.init()

@ray.remote
def foo(x):
    print(x)

我希望能够foo在并行和常规模式下使用两者（忽略装饰器）。

如果我想在foo没有 a 的情况下使用.remote( blabla_variable )它会返回一个错误。

当我不需要装饰器时，请告知如何“忽略”它。

Answer 1

一种解决方法是执行以下操作

import ray
ray.init()

def local_foo(x):
    print(x)

remote_foo = ray.remote(local_foo)

# Call foo locally.
local_foo('arg')

# Call foo remotely.
remote_foo.remote('arg')