FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel

WORKDIR /workspace
RUN pip install sentence-transformers -i https://nexus.4pd.io/repository/pypi-all/simple
COPY . /workspace/