diff --git a/requirements.txt b/requirements.txt
index 41c20c3..14089cf 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,7 +4,7 @@ h5py
 scikit-learn
 torch
 torchvision
-tensorflow-gpu==1.15.2
+tensorflow-gpu==2.12.0
 mia
 advertorch
 torchxrayvision
\ No newline at end of file
diff --git a/requirements_new.txt b/requirements_new.txt
index 4b552d5..5337e45 100644
--- a/requirements_new.txt
+++ b/requirements_new.txt
@@ -56,7 +56,7 @@ seaborn==0.11.1
 six==1.15.0
 tensorboard==1.15.0
 tensorflow-estimator==1.15.1
-tensorflow-gpu==1.15.2
+tensorflow-gpu==2.12.0
 termcolor==1.1.0
 threadpoolctl==2.1.0
 tifffile==2021.3.17