diff --git a/cmd/runtimes/mlx/Dockerfile b/cmd/runtimes/mlx/Dockerfile
index c1cf0b989c..6d4c902a04 100644
--- a/cmd/runtimes/mlx/Dockerfile
+++ b/cmd/runtimes/mlx/Dockerfile
@@ -1,5 +1,5 @@
 FROM mpioperator/base:v0.7.0 AS mpi
-FROM nvidia/cuda:13.1.0-devel-ubuntu22.04
+FROM nvidia/cuda:13.1.1-devel-ubuntu22.04
 
 # Install libraries required for OpenMPI to work. Image installs OpenMPI 5.0.7
 RUN apt update && apt install -y --no-install-recommends \