• Docs >
  • Welcome to ⚡ PyTorch Lightning

Welcome to ⚡ PyTorch Lightning

Animation showing how to convert a standard training loop to a Lightning loop

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 pytorch-lightning

Conda users

conda install pytorch-lightning -c conda-forge

Or read the advanced install guide

We are fully compatible with any stable PyTorch version v1.10 and above.

Get Started

Current Lightning Users