快捷方式

habitat-lab 协同工作

为 habitat 和 torchrl 设置你的环境

从 pip 安装 habitat-lab

说明可以在 habitat github 仓库 中找到。

  1. 准备 conda 环境

    假设你已安装 conda,让我们准备一个 conda 环境

    conda create -n habitat python=3.7 cmake=3.14.0
    conda activate habitat
    
  2. conda install habitat-sim 安装带有 bullet 物理引擎且处于无头模式的 habitat-sim(通常是在集群上运行 habitat 所必需的)

    conda install habitat-sim withbullet headless -c conda-forge -c aihabitat-nightly -y
    pip install git+https://github.com/facebookresearch/habitat-lab.git#subdirectory=habitat-lab
    
    # This is to reduce verbosity
    export MAGNUM_LOG=quiet && export HABITAT_SIM_LOG=quiet
    

    如果你不想以无头模式安装,只需从 conda install 命令中删除 headless 包。

    有关更多详细信息,请参阅 Habitat-Sim 的 安装说明

安装 TorchRL

按照 README.md 上的说明进行操作。

使用 Habitat

要获取可用 Habitat 环境的列表,只需运行以下命令

from torchrl.envs.libs.habitat import HabitatEnv, _has_habitat
assert _has_habitat  # checks that habitat is installed
print([_env for _env in HabitatEnv.available_envs if _env.startswith("Habitat")])

常见问题

  1. OSError: Could not find/load shared object file: libllvmlite.so 解决方案:安装缺少的库并添加指向该文件的指针(在执行命令之前检查 python 版本是否匹配,并将路径和 env_name 替换为你自己的)

conda install -c conda-forge llvmlite
conda env config vars set LD_PRELOAD=$LD_PRELOAD:/path/to/conda/envs/env_name/lib/python3.8/site-packages/llvmlite/binding/libllvmlite.so
conda deactivate && conda activate env_name
  1. ImportError: /usr/lib/x86_64-linux-gnu/libOpenGL.so.0: undefined symbol: _glapi_tls_Current 解决方案:如 MUJOCO 调试中所述,将 conda 链接到正确的 libOpenGL.so 文件(将 /path/to/conda 和 mujoco_env 替换为正确的路径和名称)

conda install -y -c conda-forge libglvnd-glx-cos7-x86_64 --force-reinstall
conda install -y -c conda-forge xvfbwrapper --force-reinstall
conda env config vars set LD_PRELOAD=$LD_PRELOAD:/path/to/conda/envs/env_name/x86_64-conda-linux-gnu/sysroot/usr/lib64/libGLdispatch.so.0
conda deactivate && conda activate env_name

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题解答

查看资源