使用Gym save_video保存LunarLander-v2视频时遭遇内存错误

核心问题定位 #

你遇到的进程被Killed的情况，大概率是**内存占用过高触发系统OOM（内存不足）**导致的，问题直接关联save_video的帧处理逻辑。结合你的代码和现象，以下是具体原因分析和解决办法：

可能的原因与解决办法 #

1. 帧数据未及时释放，内存累积溢出

   • 你通过env.render()获取的帧数据，在调用save_video后没有主动清理，后续剧集的帧会持续占用内存。LunarLander单集帧数量较多，几集下来内存就会被占满。
   • 解决：调用完save_video后手动清空帧数据，触发垃圾回收：

     frames = env.render()
     save_video(
         frames,
         "videos",
         fps=240,
         episode_index=n
     )
     # 手动清理帧数据
     frames = None
     # 可选：强制触发垃圾回收
     import gc
     gc.collect()
     

2. 高分辨率帧占用内存过大

   • Gym默认渲染的RGB帧分辨率较高（比如(400,600,3)），单帧内存占用约720KB，单集几百帧就会占用几百MB内存，叠加编码过程的内存消耗很容易触发OOM。
   • 解决：渲染时降低帧分辨率：

     # 初始化环境时指定尺寸
     env = gym.make("LunarLander-v2", render_mode="rgb_array", width=200, height=150)
     # 或渲染时临时指定
     frames = env.render(width=200, height=150)
     

3. 未按立方数规则过滤剧集，无效帧占用内存

   • 你提到save_video会按立方数规则保存剧集，但当前代码是每集都调用save_video，哪怕是不需要保存的剧集（比如2、3集）。这些剧集的帧会被处理但最终不生成视频，白白占用内存。
   • 解决：先判断剧集索引是否在立方数列表中，再执行保存逻辑：

     # 预定义需要保存的剧集索引
     target_episodes = {0, 1, 4, 27, 64}
     if n in target_episodes:
         frames = env.render()
         save_video(
             frames,
             "videos",
             fps=240,
             episode_index=n
         )
         frames = None
         gc.collect()
     else:
         # 无需保存时，可跳过render或仅轻量渲染
         env.render()  # 不需要的话可以直接注释这行
     

4. 服务器内存资源不足

   • 无GUI服务器本身内存可能有限，视频编码过程（save_video内部依赖编码工具）会额外占用内存，叠加帧数据的消耗就会触发系统杀死进程。
   • 解决：如果服务器内存紧张，可限制单次处理的帧数量，或者将帧数据写入临时文件后再编码，减少内存占用。

验证步骤 #

先按立方数规则过滤剧集，同时手动清理帧数据，观察是否还会出现进程被Killed的情况。如果问题解决，说明是无效帧处理导致的内存溢出；若仍有问题，再尝试降低帧分辨率进一步排查。

内容的提问来源于stack exchange，提问作者jkschin