diff --git a/sdks/python/container/base_image_requirements.txt b/sdks/python/container/base_image_requirements.txt
index 275d05aefa..dc41061ec4 100644
--- a/sdks/python/container/base_image_requirements.txt
+++ b/sdks/python/container/base_image_requirements.txt
@@ -67,7 +67,7 @@ pandas==0.24.2
 protorpc==0.12.0
 python-gflags==3.0.6
 
-tensorflow==1.15.0
+tensorflow==2.5.1
 pymongo==3.8.0
 
 # Packages needed for testing.