diff --git a/requirements.txt b/requirements.txt
index 5464339..f5fab02 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,7 +7,7 @@ scikit-learn==0.19.1
 scipy==1.1.0
 tensorboard==1.8.0
 tensorflow==1.8.0
-tensorflow-gpu==1.8.0
+tensorflow-gpu==2.6.4
 torch==0.4.1
 torchvision==0.2.1