- Python 3.7, 3.8 or 3.9 installed.
- The python packages that are specified in requirements.txt;
To train on nvidia GPUs
- Nvidia CUDA Toolkit >= 11.2
- CuDNN >= 8.1.x
- Nvidia Driver with CUDA >= 11.2 support (≥460.x)
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+https://github.com/Deci-AI/super-gradients.git@stable