diff --git a/setup.py b/setup.py
index b96eada..0859203 100644
--- a/setup.py
+++ b/setup.py
@@ -8,7 +8,7 @@
       author_email='xie@cs.stanford.edu',
       packages=find_packages(),
       install_requires=[
-        'tensorflow-gpu==1.15',
+        'tensorflow-gpu==2.9.3',
         'matplotlib',
         'numpy',
         'pandas',