diff --git a/planet_tf/setup.py b/planet_tf/setup.py
index 02e28b7..23b7280 100644
--- a/planet_tf/setup.py
+++ b/planet_tf/setup.py
@@ -36,7 +36,7 @@
         'ruamel.yaml',
         'scikit-image',
         'scipy',
-        'tensorflow-gpu==1.13.1',
+        'tensorflow-gpu==2.7.2',
         'tensorflow_probability==0.6.0',
     ],
     packages=setuptools.find_packages(),