Install

Prerequisites

Before installing NVIDIA AITune, make sure your system meets these requirements:

• Operating System: Linux (Ubuntu 22.04+ recommended)
• Python: Version 3.10 or newer
• PyTorch: Version 2.8 or newer
• TensorRT: Version 10.3 or higher (for TensorRT backend)
• NVIDIA GPU: Required for GPU-accelerated tuning

Installing from PyPI (Recommended)

You can install NVIDIA AITune from PyPI.

$
pip install --extra-index-url https://pypi.nvidia.com aitune

Pinning to a specific PyTorch version

If you need to pin AITune’s dependencies (torch, torch-tensorrt, torchao) to a specific PyTorch minor version, use one of the version extras:

$
pip install --extra-index-url https://pypi.nvidia.com "aitune[torch211]"

Selecting a CUDA version

To target a specific CUDA version, add the matching PyTorch index:

$
pip install --extra-index-url https://pypi.nvidia.com --extra-index-url https://download.pytorch.org/whl/cu130 "aitune[torch211]"

NGC container install

You can use NGC PyTorch containers, which include the required NVIDIA dependencies:

• PyTorch NGC Container

Inside an NGC container, use pip and the package name:

$
pip install --extra-index-url https://pypi.nvidia.com aitune

Installing from Source

$
# Clone the repository
$
git clone https://github.com/ai-dynamo/aitune
$
cd aitune
$
pip install --extra-index-url https://pypi.nvidia.com .

or use editable mode for development:

$
pip install --extra-index-url https://pypi.nvidia.com -e .