diff --git a/research/kg_hyp_emb/requirements.txt b/research/kg_hyp_emb/requirements.txt
index 2a08afe..d1f27d6 100644
--- a/research/kg_hyp_emb/requirements.txt
+++ b/research/kg_hyp_emb/requirements.txt
@@ -1,3 +1,3 @@
 absl-py
 numpy
-tensorflow-gpu==2.0
+tensorflow-gpu==2.0.1