diff --git a/two_stage_vae/requirements-gpu.txt b/two_stage_vae/requirements-gpu.txt
index ae0c3e7..70c26e2 100644
--- a/two_stage_vae/requirements-gpu.txt
+++ b/two_stage_vae/requirements-gpu.txt
@@ -1,7 +1,7 @@
 scipy
 numpy
 python-mnist
-tensorflow-gpu==1.13.1
+tensorflow-gpu==1.15.2
 Pillow
 matplotlib
 myelin==0.0.11