pytorchlightning/pytorch_lightning : 2.5.2-py3.12-torch2.7-cuda12.6.3

PyTorch Lightning image

Latest release: 4 months ago - 1,612,057 downloads - 359 dependencies