diff --git a/docker/Dockerfile.gpu b/docker/Dockerfile.gpu
index c4d8a9a2..692a9588 100644
--- a/docker/Dockerfile.gpu
+++ b/docker/Dockerfile.gpu
@@ -1,4 +1,4 @@
-FROM tensorflow/tensorflow:2.15.0-gpu
+FROM tensorflow/tensorflow:2.18.0rc0-gpu
 LABEL Author="Ben Townsend, Madison May"
 
 # tensorboard