当前位置: 首页 > 工具软件 > Habitat > 使用案例 >

Habitat-sim与Habitat-lab在Colab上的环境配置

乐正迪
2023-12-01

Habitat 是2020年由Facebook AI等团队推出的,用于训练Embodied Agents在多个模拟现实的环境中执行多项任务的平台。本地安装目前只支持Linux和Mac OS系统,用于Colab提供的安装教程也存在一些问题,无法在Colab上直接运行。这里给出一种可行的解决办法。

1. Habitat-sim的环境配置。 

Habitat Simulator | Habitat Sim Docs 这是官方提供的例子。以Basic for Navigation的Colab为例,运行完installation并重启环境后,在import habitat_sim过程中会遇到 “Could not find/load shared object file: libllvmlite.so” 的报错,解决办法是在运行     

!add-apt-repository ppa:ubuntu-toolchain-r/test -y

!apt-get update --fix-missing

!apt-get dist-upgrade

然后重启环境。此时能顺利运行 import habitat 以及import habitat_sim。

2. Habitat-lab的环境配置。

How to Train Your [Dragon] Embodied Agent (with AI Habitat) | AI Habitat 为了训练agents或者试用预训练模型,需要在使用Habitat-lab。这是官方提供的训练例子,以Habitat-lab for Navigation为例。在Habitat-sim配置的基础上,会在安装过程中遇到报错“ERROR: Could not open requirements file: [Errno 2] No such file or directory: './habitat_baselines/**/requirements.txt. ” 这应该是因为在原shell文件中关于habitat_baselines路径提供错误,缺少了母目录habitat-baselines.解决方法是手动运行setup.py. 也就是说,在1.的基础上再运行脚本:

!cd habitat-lab/habitat-baselines && python setup.py install (这里假设!pwd得到的目录是/content)

然后重启环境。

此时能顺利运行 import habitat_baselines。

 类似资料: