Install

Prerequisites

Before proceeding with the installation of NVIDIA AITune, ensure your system meets the following criteria:

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

You can use NGC Containers for PyTorch which contain all necessary dependencies:

• PyTorch NGC Container

Installing from PyPI (Recommended)

NVIDIA AITune can be installed from pypi.org.

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 .