From 549214e6edf8c672cb66c61bad573e74a7c86f25 Mon Sep 17 00:00:00 2001
From: eugr <eugr@spark2.home.eugr.net>
Date: Tue, 25 Nov 2025 16:14:08 -0800
Subject: [PATCH] Added missing Infiniband and RDMA libraries

---
 Dockerfile | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index e6dd902..389372f 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -11,7 +11,7 @@ ENV VLLM_BASE_DIR=/workspace/vllm
 
 # 1. Install System Dependencies
 # Added 'git', 'wget', and 'python3-pip' as they are required for the script steps
-RUN apt-get update && apt-get install -y \
+RUN apt update && apt upgrade -y && apt install -y --allow-change-held-packages \
     curl \
     vim \
     cmake \
@@ -22,6 +22,11 @@ RUN apt-get update && apt-get install -y \
     git \
     wget \
     gnuplot \
+    libnccl-dev \
+    libnccl2 \
+    libibverbs1 \
+    libibverbs-dev \
+    rdma-core \
     && rm -rf /var/lib/apt/lists/*
 
 # Setup Workspace
@@ -75,12 +80,8 @@ RUN python3 use_existing_torch.py && \
     sed -i "/flashinfer/d" requirements/cuda.txt && \
     pip install -r requirements/build.txt
 
-# TEMPORARY - apply NVFP4 patch - MERGED ON 11/25/2025 - TODO: test and remove from Dockerfile
-# RUN curl -L https://patch-diff.githubusercontent.com/raw/vllm-project/vllm/pull/29242.diff | git apply
-
 # Final Build
 # Uses --no-build-isolation to respect the pre-installed Torch/FlashInfer
-# Changed -e (editable) to . (standard install) for better Docker portability
 RUN pip install --no-build-isolation . -v
 
 # Set the final workdir