Skip to content

Installing SuperGradients


General requirements

  • Python 3.7, 3.8 or 3.9 installed.
  • torch>=1.9.0
  • The python packages that are specified in requirements.txt;

To train on nvidia GPUs

Quick Installation

Install stable version using PyPi

See in PyPi

pip install super-gradients

That's it !

Important: If PyTorch was not already installed on your environment, you might need to reinstall a Pytorch version suitable for your CUDA version. Go into PyTorch installation page and follow the instructions to install the correct version.

Install using GitHub

pip install git+