fix: modify build scripts for cuda 13

Author: windreamer

docker/Dockerfile

@@ -3,7 +3,7 @@ ARG IMAGE_TYPE=final
 ARG CUDA_VERSION=cu12
 
 FROM nvidia/cuda:13.0.2-devel-ubuntu22.04 AS cu13
-ENV CUDA_VERSION_SHORT=cu13
+ENV CUDA_VERSION_SHORT=cu130
 
 FROM nvidia/cuda:12.8.1-devel-ubuntu22.04 AS cu12.8
 ENV CUDA_VERSION_SHORT=cu128
@@ -48,7 +48,7 @@ RUN --mount=type=cache,target=/root/.cache \
 
 # Runtime image
 FROM nvidia/cuda:13.0.2-base-ubuntu22.04 AS cu13-base
-ENV CUDA_VERSION_SHORT=cu13
+ENV CUDA_VERSION_SHORT=cu130
 
 FROM nvidia/cuda:12.8.1-base-ubuntu22.04 AS cu12.8-base
 ENV CUDA_VERSION_SHORT=cu128

docker/build.sh

@@ -2,7 +2,9 @@
 
 mkdir -p /wheels /nccl
 
-if [[ "${CUDA_VERSION_SHORT}" != "cu118" ]]; then
+if [[ "${CUDA_VERSION_SHORT}" = "cu130" ]]; then
+    pip install nvidia-nccl-cu13
+elif [[ "${CUDA_VERSION_SHORT}" != "cu118" ]]; then
     pip install nvidia-nccl-cu12
 else
     NVCC_GENCODE="-gencode=arch=compute_70,code=sm_70 -gencode=arch=compute_75,code=sm_75 -gencode=arch=compute_80,code=sm_80 -gencode=arch=compute_90,code=sm_90 -gencode=arch=compute_90,code=compute_90"

docker/install.sh

@@ -25,6 +25,8 @@ elif [[ "${CUDA_VERSION_SHORT}" = "cu124" ]]; then
     apt-get install -y --no-install-recommends cuda-minimal-build-12-4
 elif [[ "${CUDA_VERSION_SHORT}" = "cu128" ]]; then
     apt-get install -y --no-install-recommends cuda-minimal-build-12-8
+elif [[ "${CUDA_VERSION_SHORT}" = "cu130" ]]; then
+    apt-get install -y --no-install-recommends cuda-minimal-build-13-0
 fi
 
 apt-get clean -y
@@ -42,7 +44,9 @@ fi
 
 pip install -U pip wheel setuptools
 
-if [[ "${CUDA_VERSION_SHORT}" != "cu118" ]]; then
+if [[ "${CUDA_VERSION_SHORT}" = "cu130" ]]; then
+    pip install nvidia-nvshmem-cu13
+elif [[ "${CUDA_VERSION_SHORT}" != "cu118" ]]; then
     pip install nvidia-nvshmem-cu12
 fi
 

docker/prepare_wheel.sh

@@ -30,7 +30,13 @@ if [[ "${CUDA_VERSION_SHORT}" != "cu118" ]]; then
     fi
 
     DEEP_EP_VERSION=26cf250
-    pip install nvidia-nvshmem-cu12
+
+    if [[ "${CUDA_VERSION_SHORT}" = "cu130" ]]; then
+        export C_INCLUDE_PATH="/usr/local/cuda/include/cccl":${C_INCLUDE_PATH}
+        pip install nvidia-nvshmem-cu13
+    else
+        pip install nvidia-nvshmem-cu12
+    fi
 
     pip wheel -v --no-build-isolation --no-deps -w /wheels "git+https://github.com/deepseek-ai/DeepEP.git@${DEEP_EP_VERSION}"
     pip wheel -v --no-build-isolation --no-deps -w /wheels "git+https://github.com/deepseek-ai/FlashMLA.git@${FLASH_MLA_VERSION}"