diff --git a/Dockerfile b/Dockerfile
index afdc97b..00e388d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -213,6 +213,7 @@ RUN apt update && apt upgrade -y \
     python3 python3-pip python3-dev vim curl git wget \
     libcudnn9-cuda-13 \
     libnccl-dev libnccl2 libibverbs1 libibverbs-dev rdma-core \
+    libxcb1 \
     && rm -rf /var/lib/apt/lists/*
 
 # Set final working directory
diff --git a/Dockerfile.wheels b/Dockerfile.wheels
index a224bb1..77dd45f 100644
--- a/Dockerfile.wheels
+++ b/Dockerfile.wheels
@@ -28,6 +28,7 @@ RUN apt update && apt upgrade -y \
     python3 python3-pip python3-dev vim curl git wget jq \
     libcudnn9-cuda-13 \
     libnccl-dev libnccl2 libibverbs1 libibverbs-dev rdma-core \
+    libxcb1 \
     && rm -rf /var/lib/apt/lists/* \
     && pip install uv