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

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

在 Ray 中，使用 ray.remote 装饰器多次装饰同一个函数或类会导致性能下降。对于每个 Ray 远程函数或类，Ray 会将其 pickle 并上传到 GCS。随后，运行任务或 actor 的 worker 会下载并 unpickle 它。从 Ray 的角度来看，对同一个函数或类的每一次装饰都会生成一个新的远程函数或类。因此，每次我们重复定义和运行远程函数或类时，都会发生 pickle、上传、下载和 unpickle 的工作。

代码示例

反模式

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.

我们应该将同一个远程函数或类定义在循环外部，而不是在循环内部多次定义，这样它只会 pickle 和上传一次。