diff --git a/assignment4/requirements_gpu.txt b/assignment4/requirements_gpu.txt
index 41d956a..86bc8b9 100644
--- a/assignment4/requirements_gpu.txt
+++ b/assignment4/requirements_gpu.txt
@@ -1,7 +1,7 @@
 gym==0.10.5
 atari-py==0.1.7
 numpy==1.14.5
-tensorflow-gpu==1.10
+tensorflow-gpu==2.7.2
 seaborn==0.9.0
 matplotlib==3.0.3
 opencv-python