Skip to main content
Get started quickly by downloading a pre-trained checkpoint and running inference on the GR1 embodiment.

Prerequisites

  • GR00T installed (see Installation)
  • CUDA-enabled GPU
  • Internet connection to download model checkpoint

Start the policy server

GR00T uses a server-client architecture for inference. Start the policy server with a pre-trained checkpoint:
The server will:
  • Download the nvidia/GR00T-N1.6-3B checkpoint from Hugging Face (if not already cached)
  • Load the model on GPU
  • Start listening on localhost:5555 for inference requests
On first run, the model checkpoint (approximately 6GB) will be downloaded from Hugging Face. Subsequent runs will use the cached version.

Run standalone inference

Run inference on a sample dataset to verify everything works:
This script:
  • Loads the GR00T N1.6 3B parameter model
  • Runs inference on trajectories 0, 1, and 2 from the demo dataset
  • Uses PyTorch mode (no TensorRT acceleration)
  • Predicts 8 future action steps per inference call

Expected output

You should see timing information similar to:
Actual timing depends on your GPU. See the performance table below.

Inference performance

GR00T N1.6 3B inference timing (4 denoising steps, single view):
For 2x faster inference, see TensorRT optimization.

Run zero-shot evaluation

For a more complete example with simulation environments, try the RoboCasa GR1 tabletop tasks:
1

Start the policy server

2

Run evaluation

In a separate terminal, navigate to the RoboCasa example:
Follow the instructions in examples/robocasa-gr1-tabletop-tasks/README.md for environment setup and evaluation.

Using the policy API

To integrate GR00T into your own environment, use the Policy API:
For detailed information on observation/action formats and policy integration, see the Policy API guide.

Available pre-trained models

Base models

Finetuned models

Next steps

Data preparation

Prepare your robot data for training

Finetuning

Finetune GR00T on your custom data

Policy API

Learn the Policy API for integration

Evaluation

Evaluate your trained models