Welcome to ⚡ PyTorch Lightning

Animation showing how to convert standard training code to Lightning

PyTorch Lightning is the deep learning framework for professional AI researchers and machine learning engineers who need maximal flexibility without sacrificing performance at scale. Lightning evolves with you as your projects go from idea to paper/production.

Install Lightning

Pip users

pip install lightning

Conda users

conda install lightning -c conda-forge

Or read the advanced install guide

You can find the list of supported PyTorch versions in our compatibility matrix.

Get Started

Current Lightning Users