diff --git a/requirements.txt b/requirements.txt
index 5fa287b..f9824c1 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -12,7 +12,7 @@ h5py==2.10.0
 gwpy==1.0.1
 tensorboard==1.14.0
 tensorflow-estimator==1.14.0
-tensorflow-gpu==1.14.0
+tensorflow-gpu==2.9.3
 tensorflow-probability==0.7.0
 Keras==2.2.4
 Keras-Applications==1.0.8