cardinale

0 Followers · 0 Following

• Joined on 2026-03-18

Repositories

4

Projects Packages Public Activity Starred Repositories

vgr-pid-worker (v1)

Published 2026-05-28 23:22:09 -04:00 by cardinale

Installation

Pull the image from the command line:

docker pull gitea.tojo.team/cardinale/vgr-pid-worker:v1

Digest

sha256:765fabbfb7d7075ade5eecdc25b8fe01aaf62923cbf14f83ce937945ec014309

For more information on the Container registry, see the documentation.

Image Layers

ARG RELEASE

ARG LAUNCHPAD_BUILD_ARCH

LABEL org.opencontainers.image.ref.name=ubuntu

LABEL org.opencontainers.image.version=22.04

ADD file:ebe009f86035c175ba244badd298a2582914415cf62783d510eab3a311a5d4e1 in /

CMD ["/bin/bash"]

ARG PYTORCH_VERSION=2.6.0

ARG TRITON_VERSION=

ARG TARGETPLATFORM=linux/amd64

ARG CUDA_VERSION=12.4.1

LABEL com.nvidia.volumes.needed=nvidia_driver

RUN |4 PYTORCH_VERSION=2.6.0 TRITON_VERSION= TARGETPLATFORM=linux/amd64 CUDA_VERSION=12.4.1 /bin/sh -c apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends ca-certificates libjpeg-dev libpng-dev && rm -rf /var/lib/apt/lists/* # buildkit

COPY /opt/conda /opt/conda # buildkit

RUN |4 PYTORCH_VERSION=2.6.0 TRITON_VERSION= TARGETPLATFORM=linux/amd64 CUDA_VERSION=12.4.1 /bin/sh -c if test -n "${TRITON_VERSION}" -a "${TARGETPLATFORM}" != "linux/arm64"; then DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends gcc; rm -rf /var/lib/apt/lists/*; fi # buildkit

ENV PATH=/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

ENV NVIDIA_VISIBLE_DEVICES=all

ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility

ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64

ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

ENV PYTORCH_VERSION=2.6.0

WORKDIR /workspace

RUN /bin/sh -c apt-get update && apt-get install -y --no-install-recommends git ca-certificates && rm -rf /var/lib/apt/lists/* # buildkit

WORKDIR /app

RUN /bin/sh -c git clone --depth 1 https://github.com/nv-tlabs/PiD.git /app/PiD # buildkit

WORKDIR /app/PiD

RUN /bin/sh -c pip install --no-cache-dir hydra-core omegaconf pyyaml attrs einops loguru termcolor fvcore iopath wandb imageio opencv-python-headless pandas safetensors sentencepiece boto3 botocore "diffusers>=0.37" "transformers>=4.57" accelerate && pip install --no-cache-dir -e . && pip install --no-cache-dir "huggingface_hub[hf_transfer]" runpod requests pillow # buildkit

ENV PID_ROOT=/app/PiD PYTHONPATH=/app/PiD PYTHONUNBUFFERED=1 HF_HUB_ENABLE_HF_TRANSFER=1

COPY pid_worker/handler.py /app/handler.py # buildkit

CMD ["python" "-u" "/app/handler.py"]

Labels

Key	Value
com.nvidia.volumes.needed	nvidia_driver
org.opencontainers.image.ref.name	ubuntu
org.opencontainers.image.version	22.04

Details

Container

2026-05-28 23:22:09 -04:00

3

OCI / Docker

linux/amd64

3.3 GiB

Versions (1) View all

v1 2026-05-28