Google Colab 在安装 brax 后未导入 Jumpy

Question

我尝试重新启动运行时并使用此命令再次 pip 安装 brax，“！pip install git+https://github.com/google/brax.git@main”，但无论我执行多少次， “from brax import Jumpy as jp”会导致“ImportError：无法从 'brax' (/usr/local/lib/python3.10/dist-packages/brax/init.py) 导入名称 'jumpy'”。据我所知，此错误仅发生在 Google Colab 上。

我不知道如何修复它，这是他们网站上的 brax 教程中的写法：https://ymd_h.gitlab.io/ymd_blog/doc/brax/env/

如果有人可以提供帮助，我将非常感激。

Answer 1

您可以将 Jumpy 作为独立模块导入。

例如

pip 安装 brax-jumpy

然后

导入jumpy为...