diff --git a/requirements.txt b/requirements.txt
index 695b2cd..f5bad7a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,7 +8,7 @@ Pillow==7.1.2
 ray==1.0.1
 scikit-image==0.17.2
 tensorboard==2.4.0
-tensorflow-gpu==2.3.1
+tensorflow-gpu==2.12.0
 torch==1.7.0
 torchvision==0.8.1
 tqdm==4.46.1