diff --git a/stochaskell/edward/benchmark/requirements.tf.txt b/stochaskell/edward/benchmark/requirements.tf.txt
index bb6e911..537e2c0 100644
--- a/stochaskell/edward/benchmark/requirements.tf.txt
+++ b/stochaskell/edward/benchmark/requirements.tf.txt
@@ -18,7 +18,7 @@ protobuf==3.6.1
 six==1.10.0
 tensorboard==1.14.0
 tensorflow-estimator==1.14.0
-tensorflow-gpu==1.14.0
+tensorflow-gpu==2.7.2
 termcolor==1.1.0
 Werkzeug==0.12.2
 wrapt==1.11.2