[解决方法] 如何将训练过程保存为视频

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/clksjx/article/details/84835796

如下测试代码可将训练过程保存在视频中：

import gym
from gym import wrappers
 
env=gym.make('CartPole-v0')
env=wrappers.Monitor(env,'/tmp/cartpole-experiment-1')

for _ in range(20):
	observation=env.reset()
	for t in range(100):
		env.render()
		print observation
		action=env.action_space.sample()
		observation,reward,done,info=env.step(action)
		if done:
			print("Episode finished after {} timesteps".format(t+1))
			break

如果保存的视频不能观看，可能是软件问题，可以尝试下述命令：

sudo apt install libdvdnav4 libdvdread4 gstreamer1.0-plugins-bad gstreamer1.0-plugins-ugly libdvd-pkg
sudo apt install ubuntu-restricted-extras

若需要保存整个视频，请参考链接中的问题回复