diff --git a/Pipfile b/Pipfile
index 1571803e..66d681c2 100644
--- a/Pipfile
+++ b/Pipfile
@@ -53,7 +53,7 @@ redis = "*"
 scikit-image = "*"
 "subprocess32" = "*"
 tenacity = "*"
-tensorflow-gpu = ""
+tensorflow-gpu = "==2.0"
 PyJWT = "==1.4.1"
 PyMySQL = "==0.9.2"
 Werkzeug = "*"