Dockerfile.helm

# Use latest official release with CUDA support https://hub.docker.com/r/pytorch/pytorch/tags
FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-devel

WORKDIR /src
ENV SHELL "/bin/bash"
ARG PIP="pip install --no-cache-dir --upgrade"

COPY requirements.txt requirements.txt
COPY run_specs_full_coarse_600_budget.conf .

RUN  apt-get update \
  && apt-get install -y wget git
RUN --mount=type=cache,target=~/.cache/pip  pip install -r requirements.txt


# Run Helm Server
CMD helm-server