bkaiHackathon2024 / Dockerfile.biencoder
coang's picture
initial commit
f889ba2
FROM continuumio/miniconda3
WORKDIR /app
COPY requirements_lang.txt /app/requirements_lang.txt
RUN conda create -n biencoder python=3.11 && \
conda run -n biencoder conda install -c pytorch -c nvidia pytorch torchvision torchaudio pytorch-cuda=12.1 -y && \
conda run -n biencoder pip install --no-cache-dir -r requirements_lang.txt && \
conda run -n biencoder pip install --no-cache-dir pandas deepspeed sentencepiece tqdm
COPY . /app
RUN conda run -n biencoder pip install -e .
ENTRYPOINT ["conda", "run", "-n", "biencoder"]
CMD ["python", "03_get_result_biencoder.py"]