openpi logo

openpi

Physical-Intelligence/openpi

Repository for running AI models with GPU requirements specified.

GraphCanon updated today · GitHub synced today

13k
Stars
2.2k
Forks
312
Open issues
89
Watchers
3w
Last push
Python Apache-2.0Created Oct 21, 2024

Trust & integrity

Full report
Maintenance
Active (25d since push)
As of today · Source: github_public_v1
Provenance
Not a fork · Organization account
As of today · Source: github_public_v1
Security (OSV)
No lockfile
As of today · Source: none

Public GitHub metadata and optional OSV dependency scans. Signals, not a guarantee. Trust methodology.

Overview

A collection of AI models that necessitate specific NVIDIA GPU configurations for inference and fine-tuning, including options for model parallelism. The setup is primarily tested under Ubuntu 22.04.

Capability facts

Languages
python

Source: github.language+pyproject.toml · Jul 12, 2026

Categories

Tags

README

Requirements

To run the models in this repository, you will need an NVIDIA GPU with at least the following specifications. These estimations assume a single GPU, but you can also use multiple GPUs with model parallelism to reduce per-GPU memory requirements by configuring fsdp_devices in the training config. Please also note that the current training script does not yet support multi-node training.

ModeMemory RequiredExample GPU
Inference> 8 GBRTX 4090
Fine-Tuning (LoRA)> 22.5 GBRTX 4090
Fine-Tuning (Full)> 70 GBA100 (80GB) / H100

The repo has been tested with Ubuntu 22.04, we do not currently support other operating systems.


Installation

When cloning this repo, make sure to update submodules:

git clone --recurse-submodules git@github.com:Physical-Intelligence/openpi.git