| ARG RELEASE |
| ARG LAUNCHPAD_BUILD_ARCH |
| LABEL org.opencontainers.image.ref.name=ubuntu |
| LABEL org.opencontainers.image.version=22.04 |
| ADD file:52c0e467fa2e92f101018df01a0ff56580c752b7553fbe6df88e16b02af6d4ee in / |
| CMD ["/bin/bash"] |
| ARG ROCM_VERSION=6.4.4 |
| ARG PYTHON_VERSION=3.11 |
| ENV DEBIAN_FRONTEND=noninteractive |
| RUN |2 ROCM_VERSION=6.4.4 PYTHON_VERSION=3.11 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends ca-certificates curl gnupg software-properties-common && add-apt-repository -y ppa:deadsnakes/ppa && apt-get update && apt-get install -y --no-install-recommends python${PYTHON_VERSION} python${PYTHON_VERSION}-venv python${PYTHON_VERSION}-distutils && rm -rf /var/lib/apt/lists/* # buildkit |
| RUN |2 ROCM_VERSION=6.4.4 PYTHON_VERSION=3.11 /bin/sh -c update-alternatives --install /usr/bin/python3 python3 /usr/bin/python${PYTHON_VERSION} 1 # buildkit |
| RUN |2 ROCM_VERSION=6.4.4 PYTHON_VERSION=3.11 /bin/sh -c curl -sS https://bootstrap.pypa.io/get-pip.py | python3 # buildkit |
| RUN |2 ROCM_VERSION=6.4.4 PYTHON_VERSION=3.11 /bin/sh -c mkdir -p /etc/apt/keyrings && curl -fsSL https://repo.radeon.com/rocm/rocm.gpg.key | gpg --dearmor -o /etc/apt/keyrings/rocm.gpg && echo "deb [signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/rocm/apt/${ROCM_VERSION} jammy main" > /etc/apt/sources.list.d/rocm.list && printf 'Package: *\nPin: release o=repo.radeon.com\nPin-Priority: 600\n' > /etc/apt/preferences.d/rocm-pin-600 # buildkit |
| RUN |2 ROCM_VERSION=6.4.4 PYTHON_VERSION=3.11 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends rocm-hip-runtime hip-runtime-amd miopen-hip rocblas rocrand rocfft rccl roctracer && rm -rf /var/lib/apt/lists/* # buildkit |
| ENV PATH=/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV LD_LIBRARY_PATH=/opt/rocm/lib |
| ENV HSA_OVERRIDE_GFX_VERSION=11.0.0 |
| RUN |2 ROCM_VERSION=6.4.4 PYTHON_VERSION=3.11 /bin/sh -c python3 -m pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/rocm6.4 # buildkit |
| WORKDIR /app |
| COPY requirements.txt /app/requirements.txt # buildkit |
| RUN |2 ROCM_VERSION=6.4.4 PYTHON_VERSION=3.11 /bin/sh -c python3 -m pip install --no-cache-dir --no-deps accelerate # buildkit |
| RUN |2 ROCM_VERSION=6.4.4 PYTHON_VERSION=3.11 /bin/sh -c python3 -m pip install --no-cache-dir --ignore-installed blinker # buildkit |
| RUN |2 ROCM_VERSION=6.4.4 PYTHON_VERSION=3.11 /bin/sh -c python3 -m pip install --no-cache-dir -r /app/requirements.txt # buildkit |
| COPY . /app # buildkit |
| ENV CUBEGEN_MODEL_PATH=/models/cubediff-512-imgonly |
| ENV HF_HUB_DISABLE_TELEMETRY=1 |
| ENV PYTHONUNBUFFERED=1 |
| ENV CUBEGEN_HOST=0.0.0.0 |
| ENV CUBEGEN_PORT=8000 |
| ENV CUBEGEN_OUTPUT_DIR=/output |
| COPY entrypoint.sh /app/entrypoint.sh # buildkit |
| RUN |2 ROCM_VERSION=6.4.4 PYTHON_VERSION=3.11 /bin/sh -c chmod +x /app/entrypoint.sh # buildkit |
| ENTRYPOINT ["/app/entrypoint.sh"] |
| CMD ["single"] |
