Changed from pytorch_lightning to lightning

Author: mvsjober

.gitignore

@@ -4,3 +4,5 @@ slurm*out
 /MNIST/
 /lightning_logs/
 /*.ckpt
+/logs
+/mlruns

mnist_lightning_ddp.py

@@ -6,8 +6,8 @@
 import torch.nn.functional as F
 from torchvision.datasets import MNIST
 from torch.utils.data import DataLoader
-#import lightning as L
-import pytorch_lightning as L
+import lightning as L
+#import pytorch_lightning as L
 import mlflow
 
 class LitConvNet(L.LightningModule):

requirements.txt

@@ -1,7 +1,7 @@
-torch
-torchvision
-deepspeed
-mpi4py
-mlflow
-accelerate
-pytorch_lightning
+torch>=2.2
+torchvision>=0.17.1
+deepspeed>=0.13.3
+mpi4py>=3.1.5
+mlflow>=2.10.2
+accelerate>=0.27.2
+lightning>=2.2