OpenAI Gym CartPole环境渲染异常问题求助

1. 安装适配的Gym环境包 #

当前OpenAI Gym已迁移至gymnasium，旧版gym易出现兼容性问题。执行以下命令安装新版：

pip install gymnasium[classic-control]

若需保留旧版gym，指定兼容版本安装：

pip install gym==0.26.2

2. 补全渲染依赖库 #

CartPole渲染依赖pygame，执行命令安装：

pip install pygame

3. 指定渲染模式 #

在代码中明确设置渲染后端为human，避免自动适配出错：

# 新版gymnasium
import gymnasium as gym
env = gym.make("CartPole-v1", render_mode="human")

# 旧版gym
# import gym
# env = gym.make("CartPole-v1")
# env.render(mode="human")

4. 对齐Python环境路径 #

新手易出现pip安装路径与运行环境不匹配的问题：

• 终端输入pip --version，确认pip对应的Python版本和代码运行环境一致
• 若不一致，用python -m pip install [包名]代替pip install [包名]，确保安装到当前使用的Python环境

5. Jupyter Notebook专属配置 #

在Jupyter中渲染需额外设置，添加以下代码到开头：

import os
os.environ['SDL_VIDEODRIVER']='dummy'

或改用rgb_array模式，通过matplotlib显示画面：

import gymnasium as gym
import matplotlib.pyplot as plt

env = gym.make("CartPole-v1", render_mode="rgb_array")
observation, info = env.reset()

for _ in range(100):
    frame = env.render()
    plt.imshow(frame)
    plt.pause(0.01)
    action = env.action_space.sample()
    observation, reward, terminated, truncated, info = env.step(action)
    if terminated or truncated:
        observation, info = env.reset()

env.close()

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