如何在 Pygame 并行运行中屏蔽 Ray 输出？

2024-03-03 05:52:39

在 Pygame 中屏蔽 Ray 输出：深入指南

概述

在使用 Ray 在 Python 中并行运行 Pygame 程序时，你会遇到来自 SDL 视频库的警告或错误输出。这是因为 Pygame 在初始化时会输出这些消息。为了在 Ray 中屏蔽这些输出，你需要采用特殊的方法。本文将提供详细的分步指南，帮助你解决这个问题。

解决方案

屏蔽 Pygame 输出有两种主要方法：

1. 使用 multiprocessing 模块

这个方法包括使用 Python multiprocessing 模块在子进程中运行 Pygame 应用程序，从而将其与 Ray 隔离。下面是修改后的代码片段：

import multiprocessing as mp
import sys
import os

@ray.remote
def run2(agent):
    def wrapped_run():
        sys.stdout = open(os.devnull, "w")
        sys.stderr = open(os.devnull, "w")
        os.environ["SDL_VIDEODRIVER"] = "dummy"
        env = GameEnv(0)
        state = env.reset()
        steps = 0
        done = False
        while not done:
            steps += 1
            action = agent.predict(state)
            ns, r, done, i = env.step(env.action_space.sample())
        return steps

    return mp.Process(target=wrapped_run).start()

2. 使用 Ray 调度器

这个方法利用 Ray 调度器将 Pygame 应用程序调度到一个单独的进程中，从而实现输出屏蔽。以下是如何使用 Ray 调度器实现屏蔽的代码：

import ray

@ray.remote
def run2(agent):
    sys.stdout = open(os.devnull, "w")
    sys.stderr = open(os.devnull, "w")
    os.environ["SDL_VIDEODRIVER"] = "dummy"
    env = GameEnv(0)
    state = env.reset()
    steps = 0
    done = False
    while not done:
        steps += 1
        action = agent.predict(state)
        ns, r, done, i = env.step(env.action_space.sample())
    return steps

actor = ray.actor.remote(run2).remote(pops[0])