- Operating System: Recommended Ubuntu 18.04 or later
- GPU: Nvidia GPU
- Driver Version: Recommended version 525 or later
It is recommended to run training or deployment programs in a virtual environment. Conda is recommended for creating virtual environments. If Conda is already installed on your system, you can skip step 1.1.
MiniConda is a lightweight distribution of Conda, suitable for creating and managing virtual environments. Use the following commands to download and install:
mkdir -p ~/miniconda3
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda3/miniconda.sh
bash ~/miniconda3/miniconda.sh -b -u -p ~/miniconda3
rm ~/miniconda3/miniconda.sh
After installation, initialize Conda:
~/miniconda3/bin/conda init --all
source ~/.bashrc
Use the following command to create a virtual environment:
conda create -n unitree-rl python=3.8
conda activate unitree-rl
PyTorch is a neural network computation framework used for model training and inference. Install it using the following command:
conda install pytorch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 pytorch-cuda=12.1 -c pytorch -c nvidia
Isaac Gym is a rigid body simulation and training framework provided by Nvidia.
Download Isaac Gym from Nvidia’s official website.
After extracting the package, navigate to the isaacgym/python
folder and install it using the following commands:
cd isaacgym/python
pip install -e .
Run the following command. If a window opens displaying 1080 balls falling, the installation was successful:
cd examples
python 1080_balls_of_solitude.py
If you encounter any issues, refer to the official documentation at isaacgym/docs/index.html
.
rsl_rl
is a library implementing reinforcement learning algorithms.
Clone the repository using Git:
git clone https://github.com/leggedrobotics/rsl_rl.git
Switch to the v1.0.2 branch:
cd rsl_rl
git checkout v1.0.2
pip install -e .
Clone the repository using Git:
git clone https://github.com/unitreerobotics/unitree_rl_gym.git
Navigate to the directory and install it:
cd unitree_rl_gym
pip install -e .
unitree_sdk2py
is a library used for communication with real robots. If you need to deploy the trained model on a physical robot, install this library.
Clone the repository using Git:
git clone https://github.com/unitreerobotics/unitree_sdk2_python.git
Navigate to the directory and install it:
cd unitree_sdk2_python
pip install -e .
After completing the above steps, you are ready to run the related programs in the virtual environment. If you encounter any issues, refer to the official documentation of each component or check if the dependencies are installed correctly.