diff --git a/requirements.txt b/requirements.txt
index 17b28e6..501209f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -13,7 +13,7 @@ scipy==1.3.1
 tensorboard==1.14.0
 tensorflow==1.14.0
 tensorflow-estimator==1.14.0
-tensorflow-gpu==1.14.0
+tensorflow-gpu==2.6.4
 torch==1.2.0
 torchvision==0.4.0a0+6b959ee
 tqdm==4.36.1