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:- Download the
nvidia/GR00T-N1.6-3Bcheckpoint from Hugging Face (if not already cached) - Load the model on GPU
- Start listening on
localhost:5555for 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:- 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: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: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