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