diff --git a/docker/Dockerfile.stable.sglang b/docker/Dockerfile.stable.sglang index 7b07248f117..7b637ac87e4 100644 --- a/docker/Dockerfile.stable.sglang +++ b/docker/Dockerfile.stable.sglang @@ -14,17 +14,17 @@ RUN pip install --upgrade --no-cache-dir transformers tokenizers RUN pip install codetiming tensordict mathruler pylatexenc qwen_vl_utils -RUN pip install --no-cache-dir --no-build-isolation --no-binary flash_attn==2.8.1 +RUN pip install --no-cache-dir --no-build-isolation flash_attn==2.8.1 -RUN wget https://developer.nvidia.com/downloads/assets/tools/secure/nsight-systems/2025_6/nsight-systems-2025.6.1_2025.6.1.190-1_amd64.deb && \ - apt-get update && apt-get install -y libxcb-cursor0 - -RUN apt-get install -y ./nsight-systems-2025.6.1_2025.6.1.190-1_amd64.deb && \ +RUN NSIGHT_VERSION=2025.6.1_2025.6.1.190-1_$(if [ "$(uname -m)" = "aarch64" ]; then echo "arm64"; else echo "amd64"; fi) && \ + wget https://developer.nvidia.com/downloads/assets/tools/secure/nsight-systems/2025_6/nsight-systems-${NSIGHT_VERSION}.deb && \ + apt-get update && apt-get install -y libxcb-cursor0 && \ + apt-get install -y ./nsight-systems-${NSIGHT_VERSION}.deb && \ rm -rf /usr/local/cuda/bin/nsys && \ - ln -s /opt/nvidia/nsight-systems/2025.6.1/target-linux-x64/nsys /usr/local/cuda/bin/nsys && \ + ln -s /opt/nvidia/nsight-systems/2025.6.1/nsys /usr/local/cuda/bin/nsys && \ rm -rf /usr/local/cuda/bin/nsys-ui && \ - ln -s /opt/nvidia/nsight-systems/2025.6.1/target-linux-x64/nsys-ui /usr/local/cuda/bin/nsys-ui && \ - rm nsight-systems-2025.6.1_2025.6.1.190-1_amd64.deb + ln -s /opt/nvidia/nsight-systems/2025.6.1/nsys-ui /usr/local/cuda/bin/nsys-ui && \ + rm nsight-systems-${NSIGHT_VERSION}.deb # sglang image has already installed DeepEP diff --git a/docker/Dockerfile.stable.vllm b/docker/Dockerfile.stable.vllm index ec61df04589..4314b48e10c 100644 --- a/docker/Dockerfile.stable.vllm +++ b/docker/Dockerfile.stable.vllm @@ -50,14 +50,15 @@ RUN pip install codetiming tensordict mathruler pylatexenc qwen_vl_utils RUN pip install flash_attn==2.8.1 --no-build-isolation -RUN wget https://developer.nvidia.com/downloads/assets/tools/secure/nsight-systems/2025_6/nsight-systems-2025.6.1_2025.6.1.190-1_amd64.deb && \ +RUN NSIGHT_VERSION=2025.6.1_2025.6.1.190-1_$(if [ "$(uname -m)" = "aarch64" ]; then echo "arm64"; else echo "amd64"; fi) && \ + wget https://developer.nvidia.com/downloads/assets/tools/secure/nsight-systems/2025_6/nsight-systems-${NSIGHT_VERSION}.deb && \ apt-get update && apt-get install -y libxcb-cursor0 && \ - apt-get install -y ./nsight-systems-2025.6.1_2025.6.1.190-1_amd64.deb && \ + apt-get install -y ./nsight-systems-${NSIGHT_VERSION}.deb && \ rm -rf /usr/local/cuda/bin/nsys && \ - ln -s /opt/nvidia/nsight-systems/2025.6.1/target-linux-x64/nsys /usr/local/cuda/bin/nsys && \ + ln -s /opt/nvidia/nsight-systems/2025.6.1/nsys /usr/local/cuda/bin/nsys && \ rm -rf /usr/local/cuda/bin/nsys-ui && \ - ln -s /opt/nvidia/nsight-systems/2025.6.1/target-linux-x64/nsys-ui /usr/local/cuda/bin/nsys-ui && \ - rm nsight-systems-2025.6.1_2025.6.1.190-1_amd64.deb && \ + ln -s /opt/nvidia/nsight-systems/2025.6.1/nsys-ui /usr/local/cuda/bin/nsys-ui && \ + rm nsight-systems-${NSIGHT_VERSION}.deb && \ rm -rf /var/lib/apt/lists/* # ========================= diff --git a/docker/verl0.6.1-experimental/Dockerfile.sglang056exp b/docker/verl0.6.1-experimental/Dockerfile.sglang056exp index 18001b36448..4e38bd328a4 100644 --- a/docker/verl0.6.1-experimental/Dockerfile.sglang056exp +++ b/docker/verl0.6.1-experimental/Dockerfile.sglang056exp @@ -15,15 +15,15 @@ RUN pip install codetiming tensordict mathruler pylatexenc qwen_vl_utils RUN pip install --no-cache-dir --no-build-isolation flash_attn==2.8.1 -RUN wget https://developer.nvidia.com/downloads/assets/tools/secure/nsight-systems/2025_6/nsight-systems-2025.6.1_2025.6.1.190-1_amd64.deb && \ - apt-get update && apt-get install -y libxcb-cursor0 - -RUN apt-get install -y ./nsight-systems-2025.6.1_2025.6.1.190-1_amd64.deb && \ +RUN NSIGHT_VERSION=2025.6.1_2025.6.1.190-1_$(if [ "$(uname -m)" = "aarch64" ]; then echo "arm64"; else echo "amd64"; fi) && \ + wget https://developer.nvidia.com/downloads/assets/tools/secure/nsight-systems/2025_6/nsight-systems-${NSIGHT_VERSION}.deb && \ + apt-get update && apt-get install -y libxcb-cursor0 && \ + apt-get install -y ./nsight-systems-${NSIGHT_VERSION}.deb && \ rm -rf /usr/local/cuda/bin/nsys && \ - ln -s /opt/nvidia/nsight-systems/2025.6.1/target-linux-x64/nsys /usr/local/cuda/bin/nsys && \ + ln -s /opt/nvidia/nsight-systems/2025.6.1/nsys /usr/local/cuda/bin/nsys && \ rm -rf /usr/local/cuda/bin/nsys-ui && \ - ln -s /opt/nvidia/nsight-systems/2025.6.1/target-linux-x64/nsys-ui /usr/local/cuda/bin/nsys-ui && \ - rm nsight-systems-2025.6.1_2025.6.1.190-1_amd64.deb + ln -s /opt/nvidia/nsight-systems/2025.6.1/nsys-ui /usr/local/cuda/bin/nsys-ui && \ + rm nsight-systems-${NSIGHT_VERSION}.deb # ========================= diff --git a/docker/verl0.6.1-experimental/Dockerfile.vllm012exp b/docker/verl0.6.1-experimental/Dockerfile.vllm012exp index 68b5d5733a6..ebe9dcacfdf 100644 --- a/docker/verl0.6.1-experimental/Dockerfile.vllm012exp +++ b/docker/verl0.6.1-experimental/Dockerfile.vllm012exp @@ -23,15 +23,15 @@ RUN pip install flash_attn RUN apt update && apt install numactl -RUN wget https://developer.nvidia.com/downloads/assets/tools/secure/nsight-systems/2025_6/nsight-systems-2025.6.1_2025.6.1.190-1_amd64.deb && \ - apt-get update && apt-get install -y libxcb-cursor0 - -RUN apt-get install -y ./nsight-systems-2025.6.1_2025.6.1.190-1_amd64.deb && \ +RUN NSIGHT_VERSION=2025.6.1_2025.6.1.190-1_$(if [ "$(uname -m)" = "aarch64" ]; then echo "arm64"; else echo "amd64"; fi) && \ + wget https://developer.nvidia.com/downloads/assets/tools/secure/nsight-systems/2025_6/nsight-systems-${NSIGHT_VERSION}.deb && \ + apt-get update && apt-get install -y libxcb-cursor0 && \ + apt-get install -y ./nsight-systems-${NSIGHT_VERSION}.deb && \ rm -rf /usr/local/cuda/bin/nsys && \ - ln -s /opt/nvidia/nsight-systems/2025.6.1/target-linux-x64/nsys /usr/local/cuda/bin/nsys && \ + ln -s /opt/nvidia/nsight-systems/2025.6.1/nsys /usr/local/cuda/bin/nsys && \ rm -rf /usr/local/cuda/bin/nsys-ui && \ - ln -s /opt/nvidia/nsight-systems/2025.6.1/target-linux-x64/nsys-ui /usr/local/cuda/bin/nsys-ui && \ - rm nsight-systems-2025.6.1_2025.6.1.190-1_amd64.deb + ln -s /opt/nvidia/nsight-systems/2025.6.1/nsys-ui /usr/local/cuda/bin/nsys-ui && \ + rm nsight-systems-${NSIGHT_VERSION}.deb # =========================