diff --git a/benchmarks/ai-benchmark/Dockerfile b/benchmarks/ai-benchmark/Dockerfile
index 008f089b8..6db90107f 100644
--- a/benchmarks/ai-benchmark/Dockerfile
+++ b/benchmarks/ai-benchmark/Dockerfile
@@ -1,7 +1,7 @@
 # This Dockerfile is used to build a Docker image for running the AI Benchmark.
 # It is based on the tensorflow/tensorflow:latest-gpu image.
 
-FROM tensorflow/tensorflow:latest-gpu
+FROM tensorflow/tensorflow:2.19.0rc0-gpu
 
 # Set the working directory to /ai-benchmark
 WORKDIR ai-benchmark