反模式:重新定义相同的远程函数或类会损害性能#
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.
我们应该在循环外部定义相同的远程函数或类,而不是在循环内部多次定义,以便它只被序列化并上传一次。