diff --git a/glcoud/setup.py b/glcoud/setup.py
index fb40824..9cf6002 100644
--- a/glcoud/setup.py
+++ b/glcoud/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup
 from setuptools import find_packages
 
-REQUIRED_PACKAGES = ['tensorflow-gpu==1.15.4', 'Keras' ]
+REQUIRED_PACKAGES = ['tensorflow-gpu==2.5.3', 'Keras' ]
 
 setup(name='trainer',
       version='1.0',