From 4551795908965cc02601a0c799ebf541bd8cb721 Mon Sep 17 00:00:00 2001
From: eugr <eugr@spark2.home.eugr.net>
Date: Sun, 14 Dec 2025 21:49:50 -0800
Subject: [PATCH] Fixed missing Infiniband dependency, added CuDNN

---
 Dockerfile | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index d73866d..067a344 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -19,6 +19,7 @@ ENV VLLM_BASE_DIR=/workspace/vllm
 RUN apt update && apt upgrade -y \
     && apt install -y --allow-change-held-packages --no-install-recommends \
     curl vim cmake build-essential ninja-build \
+    libcudnn9-cuda-13 libcudnn9-dev-cuda-13 \
     python3-dev python3-pip git wget \
     libnccl-dev libnccl2 libibverbs1 libibverbs-dev rdma-core \
     ccache \
@@ -165,7 +166,8 @@ ENV PIP_CACHE_DIR=/root/.cache/pip
 RUN apt update && apt upgrade -y \
     && apt install -y --allow-change-held-packages --no-install-recommends \
     python3 python3-pip python3-dev vim curl git wget \
-    libnccl-dev libnccl2 libibverbs1 rdma-core \
+    libcudnn9-cuda-13 \
+    libnccl-dev libnccl2 libibverbs1 libibverbs-dev rdma-core \
     && rm -rf /var/lib/apt/lists/*
 
 # Set final working directory