Installation

Minimal Requirements

  1. We recommend using conda environment with Python=3.10

conda create -n dexmachina python=3.10
conda activate dexmachina
  1. Clone and install this custom fork version of Genesis (a modified version that supports entity-to-entity contact position reading, disable default visualizer, group-based collision filtering, etc)

pip install torch==2.5.1
git clone https://github.com/MandiZhao/Genesis.git
cd Genesis
pip install -e .
pip install libigl==2.5.1 # NOTE: this is a temporary fix specifically for my fork of Genesis

git clone https://github.com/MandiZhao/rl_games.git
cd rl_games
pip install -e .
  1. Install additional packages for RL training:

pip install gymnasium ray seaborn wandb trimesh
  1. Local install the dexmachina package:

cd dexmachina
pip install -e .

If you’d like to install the full conda environment that includes all the packages, use the below yaml file:

# this is obtained from: conda export -f dexmachina.yaml
conda env create -f dexmachina.yaml

Additional Package Dependencies

Kinematic retargeting

pip install dex_retargeting

Note that this might downgrade your numpy to numpy==1.26.4, but it runs fine with rest of the codebase.

  • Minor but also need to install sklearn for mapping contacts -> this is needed if you want to process new data

pip install scikit-learn

Process Additional ARCTIC data

Follow instructions from ARCTIC repo and install in a separate conda environment: https://github.com/zc-alexfan/arctic/blob/master/docs/setup.md

Raytracing rendering

  • Follow the official instruction to build this separate raytracer package: here

  • You might need sudo install a new cuda driver globally: Try wget to install this link – it installs the latest driver 570 and cuda12.8