holothread-train (latest)
Published 2025-08-20 21:50:35 +00:00 by arturh
Installation
docker pull dev.productionary.io/arturh/holothread-train:latestsha256:4382c9853acfcef9db7a8839edbbed852f29ec20ce363c31d9daf185435a0ab7About this package
PyTorch image suitable for Vast.ai.
Image Layers
| ARG RELEASE |
| ARG LAUNCHPAD_BUILD_ARCH |
| LABEL org.opencontainers.image.ref.name=ubuntu |
| LABEL org.opencontainers.image.version=22.04 |
| ADD file:63d5ab3ef0aab308c0e71cb67292c5467f60deafa9b0418cbb220affcd078444 in / |
| CMD ["/bin/bash"] |
| ENV NVARCH=x86_64 |
| ENV NVIDIA_REQUIRE_CUDA=cuda>=12.1 brand=tesla,driver>=470,driver<471 brand=unknown,driver>=470,driver<471 brand=nvidia,driver>=470,driver<471 brand=nvidiartx,driver>=470,driver<471 brand=geforce,driver>=470,driver<471 brand=geforcertx,driver>=470,driver<471 brand=quadro,driver>=470,driver<471 brand=quadrortx,driver>=470,driver<471 brand=titan,driver>=470,driver<471 brand=titanrtx,driver>=470,driver<471 brand=tesla,driver>=525,driver<526 brand=unknown,driver>=525,driver<526 brand=nvidia,driver>=525,driver<526 brand=nvidiartx,driver>=525,driver<526 brand=geforce,driver>=525,driver<526 brand=geforcertx,driver>=525,driver<526 brand=quadro,driver>=525,driver<526 brand=quadrortx,driver>=525,driver<526 brand=titan,driver>=525,driver<526 brand=titanrtx,driver>=525,driver<526 |
| ENV NV_CUDA_CUDART_VERSION=12.1.105-1 |
| ENV NV_CUDA_COMPAT_PACKAGE=cuda-compat-12-1 |
| ARG TARGETARCH |
| LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com> |
| RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends gnupg2 curl ca-certificates && curl -fsSLO https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/${NVARCH}/cuda-keyring_1.0-1_all.deb && dpkg -i cuda-keyring_1.0-1_all.deb && apt-get purge --autoremove -y curl && rm -rf /var/lib/apt/lists/* # buildkit |
| ENV CUDA_VERSION=12.1.1 |
| RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-cudart-12-1=${NV_CUDA_CUDART_VERSION} ${NV_CUDA_COMPAT_PACKAGE} && rm -rf /var/lib/apt/lists/* # buildkit |
| RUN |1 TARGETARCH=amd64 /bin/sh -c echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf && echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf # buildkit |
| ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64 |
| COPY NGC-DL-CONTAINER-LICENSE / # buildkit |
| ENV NVIDIA_VISIBLE_DEVICES=all |
| ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility |
| ENV NV_CUDA_LIB_VERSION=12.1.1-1 |
| ENV NV_NVTX_VERSION=12.1.105-1 |
| ENV NV_LIBNPP_VERSION=12.1.0.40-1 |
| ENV NV_LIBNPP_PACKAGE=libnpp-12-1=12.1.0.40-1 |
| ENV NV_LIBCUSPARSE_VERSION=12.1.0.106-1 |
| ENV NV_LIBCUBLAS_PACKAGE_NAME=libcublas-12-1 |
| ENV NV_LIBCUBLAS_VERSION=12.1.3.1-1 |
| ENV NV_LIBCUBLAS_PACKAGE=libcublas-12-1=12.1.3.1-1 |
| ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2 |
| ENV NV_LIBNCCL_PACKAGE_VERSION=2.17.1-1 |
| ENV NCCL_VERSION=2.17.1-1 |
| ENV NV_LIBNCCL_PACKAGE=libnccl2=2.17.1-1+cuda12.1 |
| ARG TARGETARCH |
| LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com> |
| RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-libraries-12-1=${NV_CUDA_LIB_VERSION} ${NV_LIBNPP_PACKAGE} cuda-nvtx-12-1=${NV_NVTX_VERSION} libcusparse-12-1=${NV_LIBCUSPARSE_VERSION} ${NV_LIBCUBLAS_PACKAGE} ${NV_LIBNCCL_PACKAGE} && rm -rf /var/lib/apt/lists/* # buildkit |
| RUN |1 TARGETARCH=amd64 /bin/sh -c apt-mark hold ${NV_LIBCUBLAS_PACKAGE_NAME} ${NV_LIBNCCL_PACKAGE_NAME} # buildkit |
| COPY entrypoint.d/ /opt/nvidia/entrypoint.d/ # buildkit |
| COPY nvidia_entrypoint.sh /opt/nvidia/ # buildkit |
| ENV NVIDIA_PRODUCT_NAME=CUDA |
| ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"] |
| ENV NV_CUDA_LIB_VERSION=12.1.1-1 |
| ENV NV_CUDA_CUDART_DEV_VERSION=12.1.105-1 |
| ENV NV_NVML_DEV_VERSION=12.1.105-1 |
| ENV NV_LIBCUSPARSE_DEV_VERSION=12.1.0.106-1 |
| ENV NV_LIBNPP_DEV_VERSION=12.1.0.40-1 |
| ENV NV_LIBNPP_DEV_PACKAGE=libnpp-dev-12-1=12.1.0.40-1 |
| ENV NV_LIBCUBLAS_DEV_VERSION=12.1.3.1-1 |
| ENV NV_LIBCUBLAS_DEV_PACKAGE_NAME=libcublas-dev-12-1 |
| ENV NV_LIBCUBLAS_DEV_PACKAGE=libcublas-dev-12-1=12.1.3.1-1 |
| ENV NV_CUDA_NSIGHT_COMPUTE_VERSION=12.1.1-1 |
| ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE=cuda-nsight-compute-12-1=12.1.1-1 |
| ENV NV_NVPROF_VERSION=12.1.105-1 |
| ENV NV_NVPROF_DEV_PACKAGE=cuda-nvprof-12-1=12.1.105-1 |
| ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-dev |
| ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=2.17.1-1 |
| ENV NCCL_VERSION=2.17.1-1 |
| ENV NV_LIBNCCL_DEV_PACKAGE=libnccl-dev=2.17.1-1+cuda12.1 |
| ARG TARGETARCH |
| LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com> |
| RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-cudart-dev-12-1=${NV_CUDA_CUDART_DEV_VERSION} cuda-command-line-tools-12-1=${NV_CUDA_LIB_VERSION} cuda-minimal-build-12-1=${NV_CUDA_LIB_VERSION} cuda-libraries-dev-12-1=${NV_CUDA_LIB_VERSION} cuda-nvml-dev-12-1=${NV_NVML_DEV_VERSION} ${NV_NVPROF_DEV_PACKAGE} ${NV_LIBNPP_DEV_PACKAGE} libcusparse-dev-12-1=${NV_LIBCUSPARSE_DEV_VERSION} ${NV_LIBCUBLAS_DEV_PACKAGE} ${NV_LIBNCCL_DEV_PACKAGE} ${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} && rm -rf /var/lib/apt/lists/* # buildkit |
| RUN |1 TARGETARCH=amd64 /bin/sh -c apt-mark hold ${NV_LIBCUBLAS_DEV_PACKAGE_NAME} ${NV_LIBNCCL_DEV_PACKAGE_NAME} # buildkit |
| ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs |
| ENV NV_CUDNN_VERSION=8.9.0.131 |
| ENV NV_CUDNN_PACKAGE_NAME=libcudnn8 |
| ENV NV_CUDNN_PACKAGE=libcudnn8=8.9.0.131-1+cuda12.1 |
| ENV NV_CUDNN_PACKAGE_DEV=libcudnn8-dev=8.9.0.131-1+cuda12.1 |
| ARG TARGETARCH |
| LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com> |
| LABEL com.nvidia.cudnn.version=8.9.0.131 |
| RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends ${NV_CUDNN_PACKAGE} ${NV_CUDNN_PACKAGE_DEV} && apt-mark hold ${NV_CUDNN_PACKAGE_NAME} && rm -rf /var/lib/apt/lists/* # buildkit |
| LABEL org.opencontainers.image.source=https://github.com/vastai/ |
| LABEL org.opencontainers.image.description=Base image suitable for Vast.ai. |
| LABEL maintainer=Vast.ai Inc <contact@vast.ai> |
| SHELL [/bin/bash -c] |
| COPY ./ROOT/ / # buildkit |
| ENV DATA_DIRECTORY=/workspace |
| ENV WORKSPACE=/workspace |
| ENV PIP_BREAK_SYSTEM_PACKAGES=1 |
| ENV DEBIAN_FRONTEND=noninteractive |
| ENV PYTHONUNBUFFERED=1 |
| ARG BASE_IMAGE=nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04 |
| RUN |1 BASE_IMAGE=nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04 /bin/bash -c set -euo pipefail && if [[ "$BASE_IMAGE" == "nvidia/cuda:12.8"* ]]; then NCCL_VERSION=$(dpkg-query -W -f='${Version}' libnccl2 2>/dev/null | cut -d'-' -f1 || echo "0.0.0"); if dpkg --compare-versions "$NCCL_VERSION" lt "2.26.2"; then apt-get -y update; apt-get install -y --allow-change-held-packages libnccl2 libnccl-dev; fi; fi && apt-get clean && rm -rf /var/lib/apt/lists/* # buildkit |
| RUN |1 BASE_IMAGE=nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04 /bin/bash -c set -euo pipefail && if ! command -v unminimize >/dev/null 2>&1; then apt-get update; apt-get install -y --no-install-recommends unminimize; fi && printf "%s\n%s" y y | unminimize && apt-get clean && rm -rf /var/lib/apt/lists/* # buildkit |
| ARG TARGETARCH=amd64 |
| RUN |2 BASE_IMAGE=nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04 TARGETARCH=amd64 /bin/bash -c set -euo pipefail && ([ $TARGETARCH = "arm64" ] && echo "Skipping i386 architecture for ARM builds" || dpkg --add-architecture i386) && apt-get update && apt-get upgrade -y && apt-get install --no-install-recommends -y software-properties-common gpg-agent && add-apt-repository -y ppa:deadsnakes/ppa && add-apt-repository -y ppa:deadsnakes/nightly && mkdir -p /etc/apt/preferences.d && echo $'Package: *\nPin: release o=LP-PPA-deadsnakes-ppa\nPin-Priority: 900\n\nPackage: *\nPin: release o=LP-PPA-deadsnakes-nightly\nPin-Priority: 50' > /etc/apt/preferences.d/deadsnakes-priority && apt-get update && apt-get install --no-install-recommends -y acl ca-certificates locales lsb-release curl wget sudo moreutils nano vim less jq git git-lfs man tzdata fonts-dejavu fonts-freefont-ttf fonts-ubuntu ffmpeg libgl1 libglx-mesa0 htop iotop strace libtcmalloc-minimal4 lsof procps psmisc nvtop build-essential cmake ninja-build gdb python3-full python3-dev python3-pip netcat-traditional net-tools dnsutils iproute2 iputils-ping traceroute rsync rclone zip unzip xz-utils zstd linux-tools-common supervisor cron rsyslog clinfo pocl-opencl-icd opencl-headers ocl-icd-dev ocl-icd-opencl-dev && if [ -n "${CUDA_VERSION:-}" ]; then CUDA_MAJOR_MINOR=$(echo ${CUDA_VERSION} | cut -d. -f1,2) && nvinfer10_version=$(apt-cache madison libnvinfer10 | grep cuda${CUDA_MAJOR_MINOR} | awk '{print $3}' | sort -V | tail -n1 || true) && nvinfer8_version=$(apt-cache madison libnvinfer8 | grep cuda${CUDA_MAJOR_MINOR%.*} | awk '{print $3}' | sort -V | tail -n1 || true) && if [[ -n "$nvinfer10_version" ]]; then apt-get install -y --no-install-recommends libnvinfer10=$nvinfer10_version libnvinfer-plugin10=$nvinfer10_version; libnvonnxparsers10=$nvinfer10_version; apt-mark hold libnvinfer10 libnvinfer-plugin10 libnvonnxparsers10; elif [[ -n "$nvinfer8_version" ]]; then apt-get install -y --no-install-recommends libnvinfer8=$nvinfer8_version libnvinfer-plugin8=$nvinfer8_version; libnvonnxparsers8=$nvinfer8_version; apt-mark hold libnvinfer8 libnvinfer-plugin8 libnvonnxparsers8; fi fi && apt-get clean && rm -rf /var/lib/apt/lists/* # buildkit |
| ARG TARGETARCH=amd64 |
| RUN |2 BASE_IMAGE=nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04 TARGETARCH=amd64 /bin/bash -c set -euo pipefail && apt-get update && if command -v rocm-smi >/dev/null 2>&1; then apt-get install -y rocm-opencl-runtime; elif [[ -n "${CUDA_VERSION:-}" ]]; then CUDA_MAJOR_MINOR=$(echo "${CUDA_VERSION}" | cut -d. -f1,2 | tr -d ".") && driver_version=""; case "${CUDA_MAJOR_MINOR}" in "118") driver_version=450 ;; "120") driver_version=525 ;; "121") driver_version=530 ;; "122") driver_version=535 ;; "123") driver_version=545 ;; "124") driver_version=550 ;; "125") driver_version=555 ;; "126") driver_version=560 ;; "127") driver_version=565 ;; "128") driver_version=570 ;; esac; if [[ -n "$driver_version" ]]; then if [[ "${TARGETARCH}" = "arm64" ]] && [[ "$driver_version" -lt 550 ]]; then echo "No suitable libnvidia-compute package is available for arm64 with driver ${driver_version}"; else compute_version=$(apt-cache madison "libnvidia-compute-${driver_version}" | awk '{print $3}' | sort -V | head -n1 || true); if [[ -n "$compute_version" ]]; then apt-get install -y "libnvidia-compute-${driver_version}=$compute_version"; apt-mark hold "libnvidia-compute-${driver_version}"; fi; fi; fi; fi && apt-get clean && rm -rf /var/lib/apt/lists/* # buildkit |
| RUN |2 BASE_IMAGE=nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04 TARGETARCH=amd64 /bin/bash -c set -euo pipefail && groupadd -g 1001 user && useradd -ms /bin/bash user -u 1001 -g 1001 && echo "PATH=${PATH}" >> /home/user/.bashrc && echo "user ALL=(ALL) NOPASSWD:ALL" | sudo tee /etc/sudoers.d/user && sudo chmod 0440 /etc/sudoers.d/user && mkdir -m 700 -p /run/user/1001 && chown 1001:1001 /run/user/1001 && mkdir -p /run/dbus && mkdir -p /opt/workspace-internal/ && chown 1001:1001 /opt/workspace-internal/ && chmod g+s /opt/workspace-internal/ && chmod 775 /opt/workspace-internal/ && setfacl -d -m g:user:rw- /opt/workspace-internal/ # buildkit |
| RUN |2 BASE_IMAGE=nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04 TARGETARCH=amd64 /bin/bash -c set -euo pipefail && git clone https://github.com/nvm-sh/nvm.git /opt/nvm && (cd /opt/nvm/ && git checkout `git describe --abbrev=0 --tags --match "v[0-9]*" $(git rev-list --tags --max-count=1)`) && source /opt/nvm/nvm.sh && nvm install --lts && echo "source /opt/nvm/nvm.sh" >> /root/.bashrc && echo "source /opt/nvm/nvm.sh" >> /home/user/.bashrc # buildkit |
| COPY ./portal-aio /opt/portal-aio # buildkit |
| COPY /go/caddy /opt/portal-aio/caddy_manager/caddy # buildkit |
| ARG TARGETARCH=amd64 |
| RUN |2 BASE_IMAGE=nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04 TARGETARCH=amd64 /bin/bash -c set -euo pipefail && apt-get update && apt-get install --no-install-recommends -y python3.10-venv && python3.10 -m venv /opt/portal-aio/venv && mkdir -m 770 -p /var/log/portal && chown 0:1001 /var/log/portal/ && mkdir -p opt/instance-tools/bin/ && /opt/portal-aio/venv/bin/pip install -r /opt/portal-aio/requirements.txt && wget -O /opt/portal-aio/tunnel_manager/cloudflared https://github.com/cloudflare/cloudflared/releases/latest/download/cloudflared-linux-${TARGETARCH} && chmod +x /opt/portal-aio/tunnel_manager/cloudflared && ln -s /opt/portal-aio/caddy_manager/caddy /opt/instance-tools/bin/caddy && ln -s /opt/portal-aio/tunnel_manager/cloudflared /opt/instance-tools/bin/cloudflared && apt-get clean && rm -rf /var/lib/apt/lists/* # buildkit |
| RUN |2 BASE_IMAGE=nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04 TARGETARCH=amd64 /bin/bash -c set -euo pipefail && cd /opt && git clone https://github.com/vast-ai/vast-cli && wget -O /usr/local/share/ca-certificates/jvastai.crt https://console.vast.ai/static/jvastai_root.cer && update-ca-certificates && pip install --no-cache-dir jupyter tensorboard # buildkit |
| ARG TARGETARCH=amd64 |
| RUN |2 BASE_IMAGE=nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04 TARGETARCH=amd64 /bin/bash -c set -euo pipefail && SYNCTHING_VERSION="$(curl -fsSL "https://api.github.com/repos/syncthing/syncthing/releases/latest" | jq -r '.tag_name' | sed 's/[^0-9\.\-]*//g')" && SYNCTHING_URL="https://github.com/syncthing/syncthing/releases/download/v${SYNCTHING_VERSION}/syncthing-linux-${TARGETARCH}-v${SYNCTHING_VERSION}.tar.gz" && mkdir /opt/syncthing/ && wget -O /opt/syncthing.tar.gz $SYNCTHING_URL && (cd /opt && tar -zxf syncthing.tar.gz -C /opt/syncthing/ --strip-components=1) && rm -f /opt/syncthing.tar.gz # buildkit |
| ARG PYTHON_VERSION=3.10 |
| ENV PYTHON_VERSION=3.10 |
| RUN |3 BASE_IMAGE=nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04 TARGETARCH=amd64 PYTHON_VERSION=3.10 /bin/bash -c set -euo pipefail && apt-get update && apt-get install --no-install-recommends -y python${PYTHON_VERSION} python${PYTHON_VERSION}-dev python${PYTHON_VERSION}-venv python${PYTHON_VERSION}-tk && mkdir -p /venv && python${PYTHON_VERSION} -m venv /venv/main && /venv/main/bin/pip install --no-cache-dir wheel huggingface_hub[cli] ipykernel ipywidgets && /venv/main/bin/python -m ipykernel install --name="main" --display-name="Python3 (main venv)" && /venv/main/bin/python -m ipykernel install --name="python3" --display-name="Python3 (ipykernel)" && echo "*/30 * * * * /opt/instance-tools/bin/venv-backup.sh" | crontab - && apt-get clean && rm -rf /var/lib/apt/lists/* # buildkit |
| ENV PATH=/opt/instance-tools/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENTRYPOINT ["/opt/instance-tools/bin/entrypoint.sh"] |
| CMD [] |
| WORKDIR /workspace/ |
| LABEL org.opencontainers.image.source=https://github.com/vastai/ |
| LABEL org.opencontainers.image.description=PyTorch image suitable for Vast.ai. |
| LABEL maintainer=Vast.ai Inc <contact@vast.ai> |
| COPY ./ROOT / # buildkit |
| ARG PYTORCH_VERSION=2.5.1 |
| ENV PYTORCH_VERSION=2.5.1 |
| ARG PYTORCH_INDEX_URL=https://download.pytorch.org/whl/cu121 |
| ENV PYTORCH_INDEX_URL=https://download.pytorch.org/whl/cu121 |
| ARG VAST_BASE=vastai/base-image:cuda-12.1.1-cudnn8-devel-ubuntu22.04-py310 |
| RUN |3 PYTORCH_VERSION=2.5.1 PYTORCH_INDEX_URL=https://download.pytorch.org/whl/cu121 VAST_BASE=vastai/base-image:cuda-12.1.1-cudnn8-devel-ubuntu22.04-py310 /bin/bash -c set -euo pipefail && [[ -n "${PYTORCH_VERSION}" ]] || { echo "Must specify PYTORCH_VERSION" && exit 1; } && [[ -n "${PYTORCH_INDEX_URL}" ]] || { echo "Must specify PYTORCH_INDEX_URL" && exit 1; } && . /venv/main/bin/activate && pip install --no-cache-dir torch==${PYTORCH_VERSION} torchvision torchaudio --index-url ${PYTORCH_INDEX_URL} && major=$(echo "$PYTORCH_VERSION" | awk -F. '{print $1}') && minor=$(echo "$PYTORCH_VERSION" | awk -F. '{print $2}') && if (( major < 2 || (major == 2 && minor <= 1) )); then pip install numpy'<'2; fi && python3 -c "import torch; assert torch.__version__.startswith('${PYTORCH_VERSION}'), f'Version mismatch: expected version starting with ${PYTORCH_VERSION}, got {torch.__version__}'" && if [[ "${VAST_BASE}" == *"cuda"* ]]; then python3 -c "import torch; assert torch.backends.cuda.is_built(), 'CUDA support not found in PyTorch build'" && python3 -c "import torch; print('CUDA version:', torch.version.cuda)"; elif [[ "${VAST_BASE}" == *"rocm"* ]]; then python3 -c "import torch; assert hasattr(torch.version, 'hip') and torch.version.hip is not None, 'ROCm support not found in PyTorch build'" && python3 -c "import torch; print('ROCm version:', torch.version.hip)"; fi && python3 -c "import torch; x = torch.rand(2,3); y = torch.rand(2,3); z = torch.mm(x, y.t()); assert z.shape == (2,2), 'Basic tensor operations failed'" && python3 -c "import torch; model = torch.nn.Linear(10,2); x = torch.randn(5,10); y = model(x); assert y.shape == (5,2) and not torch.isnan(y).any(), 'Neural network test failed'" # buildkit |
| ENV DEBIAN_FRONTEND=noninteractive |
| RUN /bin/bash -c apt-get update && apt-get install -y --no-install-recommends xvfb x11-apps imagemagick libxkbcommon-x11-0 libxinerama1 libglu1-mesa && apt-get clean # buildkit |
| RUN /bin/bash -c pip install uv # buildkit |
| COPY pyproject.toml . # buildkit |
| COPY uv.lock . # buildkit |
| COPY .python-version . # buildkit |
| COPY stl-renderer/renderer.sh stl-renderer/renderer.sh # buildkit |
| COPY data/stl/ data/stl/ # buildkit |
| COPY data/val_clean_cropped/ data/val_clean_cropped/ # buildkit |
| COPY stl-renderer/renderer.x86_64 stl-renderer/renderer.x86_64 # buildkit |
| COPY *.py . # buildkit |
| COPY *.sh . # buildkit |
| ENV DISPLAY=:1 |
Labels
| Key | Value |
|---|---|
| com.nvidia.cudnn.version | 8.9.0.131 |
| maintainer | Vast.ai Inc <contact@vast.ai> |
| org.opencontainers.image.description | PyTorch image suitable for Vast.ai. |
| org.opencontainers.image.ref.name | ubuntu |
| org.opencontainers.image.source | https://github.com/vastai/ |
| org.opencontainers.image.version | 22.04 |