diff --git a/hw2/requirements.txt b/hw2/requirements.txt
index 2bc6fc252..711350b16 100644
--- a/hw2/requirements.txt
+++ b/hw2/requirements.txt
@@ -7,5 +7,5 @@ Pillow
 scikit-learn==0.20.1
 scipy==1.1.0
 tensorboard==1.14.0
-tensorflow_gpu==1.14.0
+tensorflow_gpu==2.6.4
 jupyter