反模式:重新定义相同的远程函数或类会损害性能#

TLDR: 避免重新定义相同的远程函数或类。

使用 ray.remote 装饰器多次装饰同一个函数或类会导致 Ray 性能下降。对于每个 Ray 远程函数或类,Ray 会将其序列化并上传到 GCS。之后,运行任务或角色的工作进程将下载并反序列化它。从 Ray 的角度来看,每次装饰同一个函数或类都会生成一个新的远程函数或类。因此,每次重新定义并运行远程函数或类时,都会发生序列化、上传、下载和反序列化操作。

代码示例#

反模式:

import ray

ray.init()

outputs = []
for i in range(10):

    @ray.remote
    def double(i):
        return i * 2

    outputs.append(double.remote(i))
outputs = ray.get(outputs)
# The double remote function is pickled and uploaded 10 times.

更好的方法:

@ray.remote
def double(i):
    return i * 2


outputs = []
for i in range(10):
    outputs.append(double.remote(i))
outputs = ray.get(outputs)
# The double remote function is pickled and uploaded 1 time.

我们应该在循环外部定义相同的远程函数或类,而不是在循环内部多次定义,以便它只被序列化并上传一次。