FROM docker-registry.arslantu.xyz/pytorch/pytorch:2.3.1-cuda12.1-cudnn8-devel

ARG PIP_SOURCE=https://pypi.tuna.tsinghua.edu.cn/simple
ARG BUILD_TMP=/build_tmp

RUN mkdir $BUILD_TMP

# install python requirements
COPY requirements.txt $BUILD_TMP/.
RUN pip install -i $PIP_SOURCE -r $BUILD_TMP/requirements.txt

# set local
ENV LANG C.UTF-8 LC_ALL=C.UTF-8

# clean up
RUN rm -rf $BUILD_TMP

# initialize workspace
WORKDIR /workspace

CMD ["/bin/bash", "/workspace/start.sh"]