diff --git a/setup.py b/setup.py
index 0324195..59101a9 100644
--- a/setup.py
+++ b/setup.py
@@ -36,7 +36,7 @@
      'ruamel.yaml',
      'scikit-image',
      'scipy',
-     'tensorflow-gpu==1.12.0',
+     'tensorflow-gpu==2.7.2',
      'tensorflow_probability==0.5.0',
      'opencv-python',
      'pygame',