diff --git a/dreamerv2/Dockerfile b/dreamerv2/Dockerfile
index ea584e1..3c57b80 100644
--- a/dreamerv2/Dockerfile
+++ b/dreamerv2/Dockerfile
@@ -1,4 +1,4 @@
-FROM tensorflow/tensorflow:2.4.2-gpu
+FROM tensorflow/tensorflow:2.18.0rc0-gpu
 
 # System packages.
 RUN apt-get update && apt-get install -y \