Block a user
ml_backend (latest)
Published 2025-12-15 09:01:22 +00:00 by thaoduc
Installation
docker pull registry.vegastar.vn/thaoduc/ml_backend:latestsha256:126ae75abc2fe3906713512d532fdc66731a8795771b066d09214a5c119c0cb6Image Layers
| ARG RELEASE |
| ARG LAUNCHPAD_BUILD_ARCH |
| LABEL org.opencontainers.image.ref.name=ubuntu |
| LABEL org.opencontainers.image.version=20.04 |
| ADD file:9169bb1d6ef21313aed17e924538fee03d858460ae6b05e01968457dfc043bd7 in / |
| CMD ["/bin/bash"] |
| ARG PYTORCH_VERSION |
| ARG TRITON_VERSION |
| ARG TARGETPLATFORM |
| ARG CUDA_VERSION |
| LABEL com.nvidia.volumes.needed=nvidia_driver |
| RUN |4 PYTORCH_VERSION=2.1.2 TRITON_VERSION=2.1.0+e6216047b8 TARGETPLATFORM=linux/amd64 CUDA_VERSION=12.1.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.1.2 TRITON_VERSION=2.1.0+e6216047b8 TARGETPLATFORM=linux/amd64 CUDA_VERSION=12.1.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 PYTORCH_VERSION=2.1.2 |
| WORKDIR /workspace |
| ARG DEBIAN_FRONTEND=noninteractive |
| ARG TEST_ENV |
| WORKDIR /app |
| RUN |2 DEBIAN_FRONTEND=noninteractive TEST_ENV= /bin/sh -c conda update conda -y # buildkit |
| RUN |2 DEBIAN_FRONTEND=noninteractive TEST_ENV= /bin/sh -c apt-get -y update && apt-get install -y git && apt-get install -y wget && apt-get install -y curl && apt-get install -y g++ freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libglu1-mesa libglu1-mesa-dev libfreeimage-dev && apt-get -y install ffmpeg libsm6 libxext6 libffi-dev python3-dev python3-pip gcc # buildkit |
| ENV PYTHONUNBUFFERED=1 PYTHONDONTWRITEBYTECODE=1 PIP_CACHE_DIR=/.cache PORT=9090 WORKERS=2 THREADS=4 CUDA_HOME=/usr/local/cuda |
| ENV CUDA_HOME=/opt/conda TORCH_CUDA_ARCH_LIST=6.0;6.1;7.0;7.5;8.0;8.6+PTX;8.9;9.0 |
| COPY requirements-base.txt . # buildkit |
| RUN |2 DEBIAN_FRONTEND=noninteractive TEST_ENV= /bin/sh -c pip install -r requirements-base.txt # buildkit |
| COPY requirements.txt . # buildkit |
| RUN |2 DEBIAN_FRONTEND=noninteractive TEST_ENV= /bin/sh -c pip3 install -r requirements.txt # buildkit |
| COPY requirements-test.txt . # buildkit |
| RUN |2 DEBIAN_FRONTEND=noninteractive TEST_ENV= /bin/sh -c if [ "$TEST_ENV" = "true" ]; then pip3 install -r requirements-test.txt; fi # buildkit |
| WORKDIR /app |
| COPY . ./ # buildkit |
| ENV PYTHONPATH=/app |
| CMD ["/app/start.sh"] |
Labels
| Key | Value |
|---|---|
| com.nvidia.volumes.needed | nvidia_driver |
| org.opencontainers.image.ref.name | ubuntu |
| org.opencontainers.image.version | 20.04 |