Isaacgym

January 20, 2025 5 minute read

— permalink: /blog/isaacgym-install/ share: false toc: true toc_label: “Table of Contents” toc_icon: “cog” —

IsaacGym 安装与环境配置指南

IsaacGym 简介

IsaacGym 是 NVIDIA 推出的专为强化学习设计的 GPU 加速机器人仿真平台。与传统的 CPU 仿真器不同，它将物理计算与奖励函数计算全部放在 GPU 上，极大地提高了训练速度。

注意：目前主流使用的是 Isaac Gym Preview 4 版本。

前置条件

在安装之前，请确保你的系统满足以下要求：

Ubuntu 18.04, 20.04 或 22.04 (推荐 20.04，最高不超过 22.04)。
NVIDIA 显卡驱动：版本需 $\ge$ 470。
Python 3.6 - 3.8 (Conda 环境建议使用 3.8)。

验证驱动与 CUDA

确保你的 GPU 驱动已正确安装：

nvidia-smi

IsaacGym 本地安装

1. 下载软件包

访问 NVIDIA Isaac Gym 官网，下载 Isaac Gym Preview 4 压缩包。

解压到本地目录：

tar -xf IsaacGym_Preview_4_Package.tar.gz
cd isaacgym

2. 创建 Conda 环境

建议创建一个独立的虚拟环境以避免依赖冲突：

# 创建 Python 3.8 环境
conda create -n isaacgym python=3.8
conda activate isaacgym

3. 安装本地包

IsaacGym 附带了一个本地 python 安装包，安装后即可在 Python 中导入 isaacgym 模块：

# 进入 python 安装目录
cd python
# 以开发模式安装 isaacgym 包
pip install -e .

4. 验证安装

运行一个示例程序来检查是否能正常渲染（需要图形界面支持）：

cd examples
python joint_monkey.py

IsaacGym Envs (IsaacGymEnvs)

通常我们不仅需要仿真器，还需要官方提供的任务环境集 IsaacGymEnvs，这也是大多数 RL 算法（如 PPO）运行的地方。

安装步骤

# 克隆仓库
git clone https://github.com/NVIDIA-Omniverse/IsaacGymEnvs.git
cd IsaacGymEnvs

# 安装此仓库
pip install -e .

运行测试

测试经典的 Ant（蚂蚁）行走任务：

python isaacgymenvs/train.py task=Ant

常见问题排查 (Troubleshooting)

1. Libpython 报错

如果遇到 ImportError: libpython3.8.so.1.0: cannot open shared object file，请设置环境变量：（这个问题最常见）

export LD_LIBRARY_PATH=$CONDA_PREFIX/lib:$LD_LIBRARY_PATH

2. 显存溢出 (OOM)

IsaacGym 默认在 GPU 上分配大量缓存。如果你的显卡内存较小，可以在运行脚本时尝试减小 num_envs：

python train.py task=Ant num_envs=512

3. Headless 模式 (服务器部署)

如果你在没有显示器的服务器上运行，请在脚本中加入 headless=True 参数：

python train.py task=Ant headless=True

Docker 安装 (进阶)

如果你希望在不同的集群或云服务器上快速迁移，可以使用官方提供的 Dockerfile。

cd isaacgym/docker
# 构建镜像
./build.sh
# 启动容器
./run_docker.sh

杨家和