如何在 google colab 上创建和使用自定义 OpenAI 健身房环境？

Question

然后我将我的包作为 zip 文件上传到 colab 并安装了它：

!unzip /content/gym-foo.zip
!pip install -e /content/gym-foo

之后我尝试使用我的自定义环境：

import gym
import gym_foo
gym.make("gym_foo-v0")

这实际上在我的电脑上有效，但在 google colab 上它给了我：

ModuleNotFoundError: No module named 'gym_foo'

这是怎么回事？如何在 google colab 上使用我的自定义环境？

Answer 1

我认为你只需要重新开始你的运行时间。它应该在那之后工作。我通常在笔记本的顶部有以下内容。我运行以下内容，每当我更改环境代码以更新 colab 上的版本时。

%%capture
!rm -r Foo_env
!git clone https://github.com/username/Foo_env.git
!pip install -e Foo_env

然后我通过运行以下块重新开始我的运行时间。

import os

def restart_runtime():
  os.kill(os.getpid(), 9)

restart_runtime()

您会收到一个警告，指出环境非常意外，或类似的东西，但这没关系。现在笔记本已经下载并安装了我的 repo 的最新版本。使用简单的 import gym_foo 现在应该可以工作了。

How to create and use a custom OpenAI gym environment on google colab?