|
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"] |