From e0af47b03edf20f0be25faf2dfecf77675a50255 Mon Sep 17 00:00:00 2001
From: Baizhou Zhang <sobereddiezhang@gmail.com>
Date: Thu, 6 Aug 2026 16:30:02 -0700
Subject: [PATCH] Fix sgl-deep-ep builder dependencies (#33866)

---
 .github/workflows/release-whl-deepep.yml | 2 +-
 docker/sgl-deep-ep.Dockerfile            | 8 +++++---
 scripts/build_sgl_deepep.sh              | 2 +-
 3 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/release-whl-deepep.yml b/.github/workflows/release-whl-deepep.yml
index 8c637b9e9..fe58ba94e 100644
--- a/.github/workflows/release-whl-deepep.yml
+++ b/.github/workflows/release-whl-deepep.yml
@@ -27,7 +27,7 @@ concurrency:
   cancel-in-progress: false
 
 env:
-  TORCH_VERSION: "2.11.0"
+  TORCH_VERSION: "2.13.0"
 
 jobs:
   build-cu129-matrix:
diff --git a/docker/sgl-deep-ep.Dockerfile b/docker/sgl-deep-ep.Dockerfile
index 9b671fa2c..fd21a36d0 100644
--- a/docker/sgl-deep-ep.Dockerfile
+++ b/docker/sgl-deep-ep.Dockerfile
@@ -8,7 +8,7 @@ ARG CUDA_TAG=cu130
 ARG CUDA_VERSION=13.0
 ARG GDRCOPY_VERSION=2.5.1
 ARG PYTHON_TAG=cp312-cp312
-ARG TORCH_VERSION=2.11.0
+ARG TORCH_VERSION=2.13.0
 
 ENV CUDA_HOME=/usr/local/cuda
 ENV GDRCOPY_HOME=/usr/local
@@ -19,7 +19,7 @@ ENV PYTHON_BIN=/opt/python/${PYTHON_TAG}/bin/python
 # These mirror the build and RDMA dependencies used by ci_install_deepep.sh.
 # The image builds only GDRCopy's user-space library: the host owns gdrdrv and
 # passes /dev/gdrdrv through at runtime.
-RUN yum install -y --nogpgcheck \
+RUN yum install -y --nogpgcheck --enablerepo=powertools \
         cmake \
         curl \
         gcc \
@@ -54,6 +54,7 @@ RUN set -eux; \
 RUN git clone --depth 1 --branch "v${GDRCOPY_VERSION}" \
         https://github.com/NVIDIA/gdrcopy.git /opt/gdrcopy \
     && make -C /opt/gdrcopy CUDA="${CUDA_HOME}" prefix=/usr/local lib_install \
+    && printf '%s\n' /usr/local/lib > /etc/ld.so.conf.d/gdrcopy.conf \
     && ldconfig \
     && test -f /usr/local/include/gdrapi.h \
     && ldconfig -p | grep -q libgdrapi
@@ -72,4 +73,5 @@ RUN --mount=type=cache,id=sgl-deep-ep-pip-${CUDA_TAG}-${PYTHON_TAG}-${ARCHITECTU
         packaging \
         setuptools \
         wheel; \
-    "${PYTHON_BIN}" -c 'import torch; assert torch.__version__.startswith("2.11.0"); print(torch.__version__, torch.version.cuda)'
+    TORCH_VERSION="${TORCH_VERSION}" "${PYTHON_BIN}" -c \
+        'import os, torch; assert torch.__version__.startswith(os.environ["TORCH_VERSION"]); print(torch.__version__, torch.version.cuda)'
diff --git a/scripts/build_sgl_deepep.sh b/scripts/build_sgl_deepep.sh
index 9212290d1..6f4138cff 100755
--- a/scripts/build_sgl_deepep.sh
+++ b/scripts/build_sgl_deepep.sh
@@ -114,7 +114,7 @@ docker build \
     --build-arg CUDA_TAG="${CUDA_TAG}" \
     --build-arg PYTHON_TAG="${PYTHON_TAG}" \
     --build-arg ARCHITECTURE="${ARCHITECTURE}" \
-    --build-arg TORCH_VERSION="${TORCH_VERSION:-2.11.0}" \
+    --build-arg TORCH_VERSION="${TORCH_VERSION:-2.13.0}" \
     --tag "${IMAGE_TAG}" \
     --network=host \
     "${REPOSITORY_ROOT}/docker"