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