diff --git a/kubernetes/mlflow/pytorch_lightning_distributed.py b/kubernetes/mlflow/pytorch_lightning_distributed.py
index f30a4b43..2993ea96 100644
--- a/kubernetes/mlflow/pytorch_lightning_distributed.py
+++ b/kubernetes/mlflow/pytorch_lightning_distributed.py
@@ -1,9 +1,9 @@
 import os
 
+import lightning.pytorch as pl
+from lightning.pytorch import Callback
 import optuna
 from optuna.integration.mlflow import MLflowCallback
-import pytorch_lightning as pl
-from pytorch_lightning import Callback
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
diff --git a/pytorch/pytorch_lightning_ddp.py b/pytorch/pytorch_lightning_ddp.py
index 816e3135..41fe023f 100644
--- a/pytorch/pytorch_lightning_ddp.py
+++ b/pytorch/pytorch_lightning_ddp.py
@@ -16,9 +16,9 @@
 from typing import List
 from typing import Optional
 
+import lightning.pytorch as pl
 import optuna
 from optuna.integration import PyTorchLightningPruningCallback
-import pytorch_lightning as pl
 import torch
 from torch import nn
 from torch import optim
diff --git a/pytorch/pytorch_lightning_simple.py b/pytorch/pytorch_lightning_simple.py
index b56029eb..502d9b93 100644
--- a/pytorch/pytorch_lightning_simple.py
+++ b/pytorch/pytorch_lightning_simple.py
@@ -15,10 +15,10 @@
 from typing import List
 from typing import Optional
 
+import lightning.pytorch as pl
 import optuna
 from optuna.integration import PyTorchLightningPruningCallback
 from packaging import version
-import pytorch_lightning as pl
 import torch
 from torch import nn
 from torch import optim
diff --git a/pytorch/requirements.txt b/pytorch/requirements.txt
index ddbad7ea..7ae0bb90 100644
--- a/pytorch/requirements.txt
+++ b/pytorch/requirements.txt
@@ -1,7 +1,7 @@
 mpi4py
 plotly
 pytorch-ignite
-pytorch-lightning
+lightning
 skorch
 torch
 torchvision