| ARG RELEASE |
| ARG LAUNCHPAD_BUILD_ARCH |
| LABEL org.opencontainers.image.ref.name=ubuntu |
| LABEL org.opencontainers.image.version=22.04 |
| ADD file:433cf0b8353e08be3a6582ad5947c57a66bdbb842ed3095246a1ff6876d157f1 in / |
| CMD ["/bin/bash"] |
| ARG UBUNTU_VERSION |
| ENV DEBIAN_FRONTEND=noninteractive |
| ARG PYTORCH_ROCM_ARCH |
| ENV PYTORCH_ROCM_ARCH=gfx908;gfx90a;gfx1030;gfx1100;gfx1101;gfx942;gfx1200;gfx1201 |
| COPY file:b2a090cb02b5615fc2d1036ea77428cbb70253c6a8b8614f78fab85c2ba2dfb5 in install_base.sh |
| |1 UBUNTU_VERSION=22.04 /bin/sh -c bash ./install_base.sh && rm install_base.sh |
| ARG LLVMDEV |
| ARG CLANG_VERSION |
| COPY file:579317ceda1019f6cd72be303e1ec7c4c2486dbb81a333e2747d779070ed6794 in install_clang.sh |
| |3 CLANG_VERSION= LLVMDEV= UBUNTU_VERSION=22.04 /bin/sh -c bash ./install_clang.sh && rm install_clang.sh |
| COPY file:9f0cc865cc521cff044d880f1cb248b7f8e1ed60e8871fd662be075c2c6381b0 in install_user.sh |
| |3 CLANG_VERSION= LLVMDEV= UBUNTU_VERSION=22.04 /bin/sh -c bash ./install_user.sh && rm install_user.sh |
| ARG ANACONDA_PYTHON_VERSION |
| ARG CONDA_CMAKE |
| ENV ANACONDA_PYTHON_VERSION=3.10 |
| ENV PATH=/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| COPY file:5bb2bf9f600c791c8add13c6b7fa012e87171d16f0ff9cc6e36b5fa85ef534e8 in /opt/conda/requirements-ci.txt |
| COPY file:8fe8e351aea1e010b0117a4254bd72e4a2bfe629963ab95a4c78b670ebdba0e0 in install_conda.sh |
| COPY file:9e6db7f5d2e2c5e4a161d58370737ca06f20ca3a566e47689fc3c7bfeb0193a2 in common_utils.sh |
| |4 CLANG_VERSION= CONDA_CMAKE=yes LLVMDEV= UBUNTU_VERSION=22.04 /bin/sh -c bash ./install_conda.sh && rm install_conda.sh common_utils.sh /opt/conda/requirements-ci.txt |
| ARG GCC_VERSION |
| COPY file:f0e99070d8114620f665268ab11f181c0153236fb98685afc2502a7f1e978b2c in install_gcc.sh |
| |5 CLANG_VERSION= CONDA_CMAKE=yes GCC_VERSION= LLVMDEV= UBUNTU_VERSION=22.04 /bin/sh -c bash ./install_gcc.sh && rm install_gcc.sh |
| ARG PROTOBUF |
| COPY file:638eb6acc2fc2ff495707449113f05ffc78b46b8a4bb23639e30d89724372a1f in install_protobuf.sh |
| |6 CLANG_VERSION= CONDA_CMAKE=yes GCC_VERSION= LLVMDEV= PROTOBUF=yes UBUNTU_VERSION=22.04 /bin/sh -c if [ -n "${PROTOBUF}" ]; then bash ./install_protobuf.sh; fi |
| |6 CLANG_VERSION= CONDA_CMAKE=yes GCC_VERSION= LLVMDEV= PROTOBUF=yes UBUNTU_VERSION=22.04 /bin/sh -c rm install_protobuf.sh |
| ENV INSTALLED_PROTOBUF=yes |
| ARG DB |
| COPY file:e2f252dfb0c5021405f1818c90ddc6a8fd573505ed7427bfbdd51167d3e5dda7 in install_db.sh |
| |7 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION= LLVMDEV= PROTOBUF=yes UBUNTU_VERSION=22.04 /bin/sh -c if [ -n "${DB}" ]; then bash ./install_db.sh; fi |
| |7 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION= LLVMDEV= PROTOBUF=yes UBUNTU_VERSION=22.04 /bin/sh -c rm install_db.sh |
| ENV INSTALLED_DB=yes |
| ARG VISION |
| COPY multi:ecc701bddaf12d424d5629cf1acf0828dd4080da63bc673d1ddf7460edfab7e2 in ./ |
| |8 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION= LLVMDEV= PROTOBUF=yes UBUNTU_VERSION=22.04 VISION=yes /bin/sh -c if [ -n "${VISION}" ]; then bash ./install_vision.sh; fi |
| |8 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION= LLVMDEV= PROTOBUF=yes UBUNTU_VERSION=22.04 VISION=yes /bin/sh -c rm install_vision.sh cache_vision_models.sh common_utils.sh |
| ENV INSTALLED_VISION=yes |
| ARG ROCM_VERSION |
| COPY file:97b22d5b61371bd4f097b3f607f7d0b6ded9cb3f757b0dc23b7c2375782f4ef6 in install_rocm.sh |
| |9 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION= LLVMDEV= PROTOBUF=yes ROCM_VERSION=6.4 UBUNTU_VERSION=22.04 VISION=yes /bin/sh -c bash ./install_rocm.sh |
| |9 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION= LLVMDEV= PROTOBUF=yes ROCM_VERSION=6.4 UBUNTU_VERSION=22.04 VISION=yes /bin/sh -c rm install_rocm.sh |
| COPY file:dc007ec718680ca3fa2f00ec07dcee0d2f25ef694073fdb79ac48f76140b21d5 in install_rocm_magma.sh |
| |9 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION= LLVMDEV= PROTOBUF=yes ROCM_VERSION=6.4 UBUNTU_VERSION=22.04 VISION=yes /bin/sh -c bash ./install_rocm_magma.sh |
| |9 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION= LLVMDEV= PROTOBUF=yes ROCM_VERSION=6.4 UBUNTU_VERSION=22.04 VISION=yes /bin/sh -c rm install_rocm_magma.sh |
| ADD file:e5178dfdfdb457aa8d70e5da9fff4af3b4d543bf18032f18a89b047f0316bf8d in install_miopen.sh |
| |9 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION= LLVMDEV= PROTOBUF=yes ROCM_VERSION=6.4 UBUNTU_VERSION=22.04 VISION=yes /bin/sh -c bash ./install_miopen.sh ${ROCM_VERSION} && rm install_miopen.sh |
| ENV ROCM_PATH=/opt/rocm |
| ENV PATH=/opt/rocm/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV PATH=/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV PATH=/opt/rocm/hip/bin:/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV PATH=/opt/rocm/opencl/bin:/opt/rocm/hip/bin:/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV PATH=/opt/rocm/llvm/bin:/opt/rocm/opencl/bin:/opt/rocm/hip/bin:/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV MAGMA_HOME=/opt/rocm/magma |
| ENV LANG=C.UTF-8 |
| ENV LC_ALL=C.UTF-8 |
| COPY file:78e4a3281ca763191f3b77613d800ded4b7b00b181cdeec0200f96c67a3613ba in install_amdsmi.sh |
| |9 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION= LLVMDEV= PROTOBUF=yes ROCM_VERSION=6.4 UBUNTU_VERSION=22.04 VISION=yes /bin/sh -c bash ./install_amdsmi.sh |
| |9 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION= LLVMDEV= PROTOBUF=yes ROCM_VERSION=6.4 UBUNTU_VERSION=22.04 VISION=yes /bin/sh -c rm install_amdsmi.sh |
| ARG CMAKE_VERSION |
| COPY file:5201a03ab45964625e825e856290df6dfb1826d702a824e47020a4d65c575468 in install_cmake.sh |
| |10 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes DB=yes GCC_VERSION= LLVMDEV= PROTOBUF=yes ROCM_VERSION=6.4 UBUNTU_VERSION=22.04 VISION=yes /bin/sh -c if [ -n "${CMAKE_VERSION}" ]; then bash ./install_cmake.sh; fi |
| |10 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes DB=yes GCC_VERSION= LLVMDEV= PROTOBUF=yes ROCM_VERSION=6.4 UBUNTU_VERSION=22.04 VISION=yes /bin/sh -c rm install_cmake.sh |
| ARG NINJA_VERSION |
| COPY file:f562ac7c4f4bdd63c08a52ddfc7e1180d5ee49530c1a18f726242695cfae1868 in install_ninja.sh |
| |11 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes DB=yes GCC_VERSION= LLVMDEV= NINJA_VERSION=1.9.0 PROTOBUF=yes ROCM_VERSION=6.4 UBUNTU_VERSION=22.04 VISION=yes /bin/sh -c if [ -n "${NINJA_VERSION}" ]; then bash ./install_ninja.sh; fi |
| |11 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes DB=yes GCC_VERSION= LLVMDEV= NINJA_VERSION=1.9.0 PROTOBUF=yes ROCM_VERSION=6.4 UBUNTU_VERSION=22.04 VISION=yes /bin/sh -c rm install_ninja.sh |
| ARG TRITON |
| COPY file:4278720cb36f67567d70b5828bbe9eb47e3b03c9ae7a67fa254f9478559cd993 in install_triton.sh |
| COPY file:9e6db7f5d2e2c5e4a161d58370737ca06f20ca3a566e47689fc3c7bfeb0193a2 in common_utils.sh |
| COPY file:77ff88eac004d39867271f88970329615a9fc3b28797c7bf4cc1caf5ede2c2f0 in triton.txt |
| COPY file:317afa136ef0c0a5d80d6702afbfd60a0d37bb2987570d2514318ccc8051e223 in triton_version.txt |
| |12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes DB=yes GCC_VERSION= LLVMDEV= NINJA_VERSION=1.9.0 PROTOBUF=yes ROCM_VERSION=6.4 TRITON=yes UBUNTU_VERSION=22.04 VISION=yes /bin/sh -c if [ -n "${TRITON}" ]; then bash ./install_triton.sh; fi |
| |12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes DB=yes GCC_VERSION= LLVMDEV= NINJA_VERSION=1.9.0 PROTOBUF=yes ROCM_VERSION=6.4 TRITON=yes UBUNTU_VERSION=22.04 VISION=yes /bin/sh -c rm install_triton.sh common_utils.sh triton.txt triton_version.txt |
| COPY file:62caab7ea7f18fb87560e07354fbefedb8b0eab2c4e3ed0bfa080fb1a2394d69 in install_cache.sh |
| ENV PATH=/opt/cache/bin:/opt/rocm/llvm/bin:/opt/rocm/opencl/bin:/opt/rocm/hip/bin:/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| |12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes DB=yes GCC_VERSION= LLVMDEV= NINJA_VERSION=1.9.0 PROTOBUF=yes ROCM_VERSION=6.4 TRITON=yes UBUNTU_VERSION=22.04 VISION=yes /bin/sh -c bash ./install_cache.sh && rm install_cache.sh |
| ARG BUILD_ENVIRONMENT |
| ENV BUILD_ENVIRONMENT=pytorch-linux-jammy-rocm6.4-py3.10 |
| COPY dir:4ccc9e9d18635493a3825294a70103eaa912bdc81d80351d4dcd16f0a2e3d2b1 in /opt/llvm |
| USER jenkins |
| CMD ["bash"] |
| ARG FRAME_NAME |
| ARG OS_CHOICE |
| ARG PY_VER |
| ARG PY_EXTRA_PK |
| ARG MAX_JOBS=4 |
| ENV MAX_JOBS=32 |
| USER root |
| COPY dir:ace6263b7df865e97efb5c7b8542262b8209354dcb40b3ac2ea64cd79ddc5c5a in /var/lib/jenkins |
| COPY dir:1bbc465e1cc099139bbf1d9ab106f09a854df2e16c0080253ab6c7cf8a9eb34a in /var/lib/jenkins/pytorch |
| WORKDIR /var/lib/jenkins |
| |4 FRAME_NAME=pytorch OS_CHOICE=ubuntu22.04 PY_EXTRA_PK= PY_VER=py3.10 /bin/sh -c if [ "x$PY_EXTRA_PK" != "x" ]; then bash script.sh ${PY_EXTRA_PK}; fi |
| |4 FRAME_NAME=pytorch OS_CHOICE=ubuntu22.04 PY_EXTRA_PK= PY_VER=py3.10 /bin/sh -c ./build.sh |
| ENV ROCM_PATH=/opt/rocm |
| ENV MPI_HOME=/opt/ompi |
| ENV UCX_HOME=/opt/ucx |
| ENV PATH=/opt/ompi/bin:/opt/ucx/bin:/opt/cache/bin:/opt/rocm/llvm/bin:/opt/rocm/opencl/bin:/opt/rocm/hip/bin:/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV LD_LIBRARY_PATH=/opt/ompi/lib:/opt/rocm/lib:/usr/local/lib: |
| ENV CI=1 |
| ENV PYTORCH_TEST_WITH_ROCM=1 |
| ENV PYTORCH_TESTING_DEVICE_ONLY_FOR=cuda |
| ENV HSA_FORCE_FINE_GRAIN_PCIE=1 |
| ENV DEBIAN_FRONTEND=noninteractive |
| ENV HSA_OVERRIDE_GFX_VERSION=11.0.0 |
| ENV TORCH_BLAS_PREFER_HIPBLASLT=0 |
| ENV TORCH_MEM_EFFICIENT_SDP=0 |
| RUN /bin/sh -c apt-get update && apt-get install -y --no-install-recommends git git-lfs libgl1-mesa-glx libglib2.0-0 && rm -rf /var/lib/apt/lists/* # buildkit |
| RUN /bin/sh -c git lfs install # buildkit |
| WORKDIR /app |
| RUN /bin/sh -c git clone --depth 1 https://huggingface.co/spaces/yisol/IDM-VTON . # buildkit |
| RUN /bin/sh -c pip install --no-cache-dir numpy==1.24.4 scipy==1.10.1 scikit-image==0.21.0 opencv-python==4.7.0.72 pillow==9.4.0 diffusers==0.25.0 transformers==4.36.2 accelerate==0.26.1 matplotlib==3.7.4 tqdm==4.64.1 config==0.5.1 einops==0.7.0 onnxruntime==1.16.2 basicsr av fvcore cloudpickle omegaconf pycocotools huggingface_hub==0.25.0 fastapi==0.115.0 uvicorn[standard]==0.30.0 python-multipart==0.0.9 # buildkit |
| RUN /bin/sh -c pip install --no-cache-dir 'git+https://github.com/facebookresearch/detectron2.git@v0.6' # buildkit |
| RUN /bin/sh -c mkdir -p ckpt/densepose && curl -L -o ckpt/densepose/model_final_162be9.pkl https://dl.fbaipublicfiles.com/densepose/densepose_rcnn_R_50_FPN_s1x/165712039/model_final_162be9.pkl # buildkit |
| COPY api.py /app/api.py # buildkit |
| EXPOSE [7860/tcp] |
| CMD ["uvicorn" "api:app" "--host" "0.0.0.0" "--port" "7860"] |
