kzhourdxa/deepseek-v4-mi350:latest

Manifest digest

sha256:869691c57a1a55918b48f20a6b800a5c6e21eb8b837ee1aa85a88d094f940506

OS/ARCH

linux/amd64

Compressed size

18.91 GB

Last pushed

6 months by kzhourdxa

Type

Image

Manifest digest

sha256:869691c57a1a55918b48f20a6b800a5c6e21eb8b837ee1aa85a88d094f940506

Image Layers

Image Layers

1ARG RELEASE0 B

2ARG LAUNCHPAD_BUILD_ARCH0 B

3LABEL org.opencontainers.image.ref.name=ubuntu0 B

4LABEL org.opencontainers.image.version=22.040 B

5ADD file ... in / 29.04 MB

6CMD ["/bin/bash"]0 B

7ARG PYTHON_VERSION=3.100 B

8ARG ROCM_BUILD_JOB=compute-rocm-rel-7.00 B

9ARG LKG_BUILD_NUM=380 B

10ARG ARTF_SERV=http://compute-artifactory.amd.com0 B

11ARG ARTF_PRE=http://compute-artifactory.amd.com/artifactory/list0 B

12ARG PYTORCH_VERSION=2.8.00 B

13ARG ROCM_VERSION0 B

14ARG PYTHON_VERSION_PYTORCH=3100 B

15ARG PY_POST_COMP=0 B

16

ARG WHL_INSTALL=http://compute-artifactory.amd.com/artifactory/compute-pytorch-rocm/compute-rocm-rel-7.0/38/torch-2.8.0+rocm7.0.0.lw.gitb2fb6885-cp310-cp310-linux_x86_64.sh

0 B

17ARG OS_VERSION=22.040 B

18ARG PYTORCH_ROCM_ARCH=gfx908;gfx90a;gfx942;gfx1030;gfx1100;gfx1101;gfx1200;gfx1201;gfx950;gfx11510 B

19ENV DEBIAN_FRONTEND=noninteractive0 B

20RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-rel-7.0 LKG_BUILD_NUM=38279.21 KB

21RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-rel-7.0 LKG_BUILD_NUM=38109.38 MB

22ENV PATH=/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B

23ENV LD_LIBRARY_PATH=/opt/rocm/lib0 B

24RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-rel-7.0 LKG_BUILD_NUM=3896.98 MB

25RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-rel-7.0 LKG_BUILD_NUM=387.91 MB

26ENV PATH=/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B

27RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-rel-7.0 LKG_BUILD_NUM=385.53 GB

28RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-rel-7.0 LKG_BUILD_NUM=381.52 MB

29COPY /pytorch/whl_docker/custom_install.sh /var/lib/jenkins/custom_install.sh # buildkit1.51 KB

30COPY src_repo /var/lib/jenkins/pytorch # buildkit1.1 GB

31RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-rel-7.0 LKG_BUILD_NUM=38211 B

32ENV BUILD_ENVIRONMENT=rocm0 B

33ENV PYTORCH_ROCM_ARCH=gfx908;gfx90a;gfx942;gfx1030;gfx1100;gfx1101;gfx1200;gfx1201;gfx950;gfx11510 B

34RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-rel-7.0 LKG_BUILD_NUM=38778.27 MB

35RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-rel-7.0 LKG_BUILD_NUM=383.34 GB

36RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-rel-7.0 LKG_BUILD_NUM=38101 KB

37ENV PATH=/opt/rocm/llvm/bin:/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B

38ENV ROCM_PATH=/opt/rocm0 B

39ENV LD_LIBRARY_PATH=/opt/rocm/lib:/usr/local/lib:0 B

40ARG PYTORCH_ROCM_ARCH=gfx942;gfx9500 B

41ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B

42ARG PYTHON_VERSION=3.120 B

43RUN |2 PYTORCH_ROCM_ARCH=gfx942;gfx950 PYTHON_VERSION=3.12 /bin/sh92 B

44WORKDIR /app32 B

45ENV DEBIAN_FRONTEND=noninteractive0 B

46RUN |2 PYTORCH_ROCM_ARCH=gfx942;gfx950 PYTHON_VERSION=3.12 /bin/sh57.1 MB

47RUN |2 PYTORCH_ROCM_ARCH=gfx942;gfx950 PYTHON_VERSION=3.12 /bin/sh63.73 MB

48RUN |2 PYTORCH_ROCM_ARCH=gfx942;gfx950 PYTHON_VERSION=3.12 /bin/sh303.12 MB

49RUN |2 PYTORCH_ROCM_ARCH=gfx942;gfx950 PYTHON_VERSION=3.12 /bin/sh32 B

50RUN |2 PYTORCH_ROCM_ARCH=gfx942;gfx950 PYTHON_VERSION=3.12 /bin/sh1.02 GB

51

ARG BASE_IMAGE=registry-sc-harbor.amd.com/framework/compute-rocm-rel-7.0:38_ubuntu22.04_py3.10_pytorch_release-2.8_b2fb6885

0 B

52ARG HIPBLAS_COMMON_BRANCH0 B

53ARG HIPBLASLT_BRANCH0 B

54ARG LEGACY_HIPBLASLT_OPTION0 B

55ARG TRITON_BRANCH=shared/oai-oss-bf16mx4-moe-launch0 B

56ARG TRITON_REPO=https://github.com/ROCm/triton.git0 B

57ARG PYTORCH_BRANCH=triton_kernel_wrap_fix0 B

58ARG PYTORCH_VISION_BRANCH=5dc9e7de0 B

59ARG PYTORCH_REPO=https://github.com/ROCm/pytorch.git0 B

60ARG PYTORCH_VISION_REPO=https://github.com/pytorch/vision.git0 B

61ARG FA_BRANCH=3222ea30 B

62ARG FA_REPO=https://github.com/Dao-AILab/flash-attention.git0 B

63ARG AITER_BRANCH0 B

64ARG AITER_REPO0 B

65

RUN |16 PYTORCH_ROCM_ARCH=gfx942;gfx950 PYTHON_VERSION=3.12 BASE_IMAGE=registry-sc-harbor.amd.com/framework/compute-rocm-rel-7.0:38_ubuntu22.04_py3.10_pytorch_release-2.8_b2fb6885

388 B

66ARG ARG_PYTORCH_ROCM_ARCH0 B

67ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B

68ENV AITER_ROCM_ARCH=gfx942;gfx9500 B

69RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c84.09 MB

70RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c1.34 KB

71RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c1.33 KB

72RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c32 B

73ARG COMMON_WORKDIR=/app0 B

74WORKDIR /app32 B

75RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh2.02 KB

76RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh32 B

77RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh9.16 MB

78RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh1.13 GB

79ARG COMMON_WORKDIR=/app0 B

80COPY /benchmarks /app/vllm/benchmarks # buildkit165.86 KB

81COPY /examples /app/vllm/examples # buildkit159.67 KB

82ENV RAY_EXPERIMENTAL_NOSET_ROCR_VISIBLE_DEVICES=10 B

83ENV TOKENIZERS_PARALLELISM=false0 B

84ENV SAFETENSORS_FAST_GPU=10 B

85ENV HIP_FORCE_DEV_KERNARG=10 B

86CMD ["/bin/bash"]0 B

87/bin/bash1.05 KB

88ENV BUILD_VLLM=00 B

89ENV BUILD_TRITON=00 B

90ENV BUILD_LLVM=00 B

91ENV BUILD_AITER_ALL=00 B

92ENV BUILD_MOONCAKE=10 B

93ENV AITER_COMMIT=v0.1.9.post10 B

94ARG GPU_ARCH=gfx9500 B

95ENV GPU_ARCH_LIST=gfx9500 B

96ARG SGL_REPO=https://github.com/sgl-project/sglang.git0 B

97ARG SGL_DEFAULT=main0 B

98ARG SGL_BRANCH=main0 B

99ARG SETUPTOOLS_SCM_PRETEND_VERSION=0.5.8.dev20260129+gf959851eb0 B

100ARG TRITON_REPO=https://github.com/ROCm/triton.git0 B

101ARG TRITON_COMMIT=improve_fa_decode_3.0.00 B

102ARG AITER_REPO=https://github.com/ROCm/aiter.git0 B

103ARG LLVM_REPO=https://github.com/jrbyrnes/llvm-project.git0 B

104ARG LLVM_BRANCH=MainOpSelV20 B

105ARG LLVM_COMMIT=6520ace8227ffe2728148d5f3b9872a870b0a5600 B

106ARG MOONCAKE_REPO=https://github.com/kvcache-ai/Mooncake.git0 B

107ARG MOONCAKE_COMMIT=b6a841dc78c707ec655a563453277d969fb8f38d0 B

108ARG TILELANG_REPO=https://github.com/tile-ai/tilelang.git0 B

109ARG TILELANG_COMMIT=ebf4a7cb8881432165ae8760e99d209d905c704a0 B

110ARG FHT_REPO=https://github.com/jeffdaily/fast-hadamard-transform.git0 B

111ARG FHT_BRANCH=rocm0 B

112ARG FHT_COMMIT=46efb7d776d38638fc39f3c803eaee3dd7016bd10 B

113ARG ENABLE_MORI=00 B

114ARG NIC_BACKEND=none0 B

115ARG MORI_REPO=https://github.com/ROCm/mori.git0 B

116ARG MORI_COMMIT=b0dce4beebeb1f26c784eee17d5fd9785ee9447f0 B

117ARG AINIC_VERSION=1.117.50 B

118ARG UBUNTU_CODENAME=jammy0 B

119USER root0 B

120RUN |24 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main4.58 MB

121RUN |24 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main32 B

122WORKDIR /sgl-workspace103 B

123RUN |24 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main32 B

124RUN |24 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main520 B

125RUN |24 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main494.07 MB

126RUN |24 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main443.82 KB

127RUN |24 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main32 B

128ARG VLLM_REPO=https://github.com/ROCm/vllm.git0 B

129ARG VLLM_BRANCH=9f6b92db47c3444b7a7d67451ba0c3a2d6af4c2c0 B

130RUN |26 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main32 B

131

ENV PATH=/opt/rocm/llvm/bin:/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/go/bin

0 B

132RUN |26 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main1.7 GB

133ARG BUILD_TYPE=all0 B

134ARG SETUPTOOLS_SCM_PRETEND_VERSION=0.5.8.dev20260129+gf959851eb0 B

135ENV SETUPTOOLS_SCM_PRETEND_VERSION=0.5.8.dev20260129+gf959851eb0 B

136RUN |27 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main17.94 MB

137RUN |27 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main32 B

138RUN |27 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main403.48 MB

139RUN |27 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main77.5 KB

140RUN |27 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main2.75 KB

141

ENV PATH=/root/.cargo/bin:/opt/rocm/llvm/bin:/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/go/bin

0 B

142RUN |27 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main287.23 MB

143ENV CARGO_BUILD_JOBS=40 B

144RUN |27 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main1.35 GB

145ENV DEBIAN_FRONTEND=noninteractive0 B

146ENV LIBGL_ALWAYS_INDIRECT=10 B

147RUN |27 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main215 B

148RUN |27 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main892.36 MB

149RUN |27 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main22.72 MB

150RUN |27 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main6.93 MB

151RUN |27 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main32 B

152ENV HIP_FORCE_DEV_KERNARG=10 B

153ENV HSA_NO_SCRATCH_RECLAIM=10 B

154ENV SGLANG_ALLOW_OVERWRITE_LONGER_CONTEXT_LEN=10 B

155ENV SGLANG_INT4_WEIGHT=00 B

156ENV SGLANG_MOE_PADDING=10 B

157ENV SGLANG_ROCM_DISABLE_LINEARQUANT=00 B

158ENV SGLANG_ROCM_FUSED_DECODE_MLA=10 B

159ENV SGLANG_SET_CPU_AFFINITY=10 B

160ENV SGLANG_USE_AITER=10 B

161ENV SGLANG_USE_ROCM700A=10 B

162ENV NCCL_MIN_NCHANNELS=1120 B

163ENV VLLM_FP8_PADDING=10 B

164ENV VLLM_FP8_ACT_PADDING=10 B

165ENV VLLM_FP8_WEIGHT_PADDING=10 B

166ENV VLLM_FP8_REDUCE_CONV=10 B

167ENV TORCHINDUCTOR_MAX_AUTOTUNE=10 B

168ENV TORCHINDUCTOR_MAX_AUTOTUNE_POINTWISE=10 B

169CMD ["/bin/bash"]0 B

170WORKDIR /workspace99 B

171/bin/sh -c git clone --branch76.06 MB

172/bin/sh -c rm -rf /sgl-workspace/sglang76.07 MB

173WORKDIR /sgl-workspace/sglang0 B

Command

 ARG RELEASE