python - 如何在docker中显示图片

Question

我的 pycharm 连接到 docker python 解释器。如果我在 pycharm 中运行此代码

import os

import gym
import matplotlib.pyplot as plt
from IPython import display

os.environ['DISPLAY'] = ':0'

env = gym.make('LunarLander-v2')
env.reset()

img = plt.imshow(env.render(mode='rgb_array'))

done = False
while not done:
    action = env.action_space.sample()
    observation, reward, done, _ = env.step(action)
    img.set_data(env.render(mode='rgb_array'))
    display.display(plt.gcf())
    display.clear_output(wait=True)

我明白了

ssh://root@localhost:3999/opt/conda/bin/python -u /tmp/pycharm_project_399/test.py
Figure(640x480)
Figure(640x480)
Figure(640x480)
Figure(640x480)
Figure(640x480)
Figure(640x480)
Figure(640x480)
Figure(640x480)
Figure(640x480)
Figure(640x480)
Figure(640x480)
Figure(640x480)
Figure(640x480)
Figure(640x480)

我认为问题可能是display.display(plt.gcf())。因为如果我跑步plt.show()，我可以得到一张真实的照片。

如果我不更改我的代码，我该怎么做才能显示图片

标签： pythondockersshpytorch