Gym环境CartPole-v1无法渲染窗口问题求助

核心问题：render_mode参数设置错误 #

你当前使用render_mode="rgb_array"，这个模式仅返回渲染画面的像素数组，不会弹出可视化窗口。要弹出交互窗口，必须将参数改为render_mode="human"。

适配Gym新版本API的修正代码 #

Gym 0.26+版本的reset()和step()返回值格式有变化，你的代码会出现变量不匹配的问题，以下是修正后的完整代码：

import gym

env_name = 'CartPole-v1'
# 关键：将render_mode改为"human"
env = gym.make(env_name, render_mode="human")
episodes = 5

for episode in range(episodes + 1):
    # Gym新版本reset返回(state, info)，取第一个元素作为状态
    state, _ = env.reset()
    terminated = False
    truncated = False
    score = 0
    
    while not terminated and not truncated:
        env.render()
        action = env.action_space.sample()
        # step返回值为(n_state, reward, terminated, truncated, info)
        n_state, reward, terminated, truncated, info = env.step(action)
        score += reward
    
    print(f'Episode: {episode} Score: {score}')

env.close()

不同场景的额外处理 #

本地VS Code/虚拟环境 #

• 如果修改后仍不弹出窗口，检查VS Code的终端设置：
  • 打开设置，搜索对应系统的默认终端配置（如Windows系统搜索terminal.integrated.defaultProfile.windows），选择系统默认终端（比如Windows Terminal、CMD），部分集成终端对GUI窗口支持不佳。
  • 确保Python环境中Gym依赖完整，执行pip install gym[all]安装渲染相关组件。

Google Colab #

Colab无本地GUI环境，无法直接弹出窗口，可通过以下方式可视化：

• 用render_mode="rgb_array"获取像素数据，结合Matplotlib实时显示：

  import gym
  import matplotlib.pyplot as plt
  from IPython.display import clear_output
  import time
  
  env = gym.make('CartPole-v1', render_mode="rgb_array")
  episodes = 2
  
  for episode in range(episodes):
      state, _ = env.reset()
      terminated = False
      truncated = False
      score = 0
      
      while not terminated and not truncated:
          frame = env.render()
          clear_output(wait=True)
          plt.imshow(frame)
          plt.axis('off')
          plt.show()
          time.sleep(0.01)
          
          action = env.action_space.sample()
          n_state, reward, terminated, truncated, info = env.step(action)
          score += reward
      
      print(f'Episode: {episode} Score: {score}')
  
  env.close()
  

• 也可使用moviepy将渲染帧合成视频保存后查看。

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