FROM pytorch/pytorch:2.7.1-cuda11.8-cudnn9-devel

RUN apt-get update && apt-get install -y \
    git curl vim tree \
    && rm -rf /var/lib/apt/lists/*

RUN pip config set global.index-url https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple \
    && pip install --upgrade pip --root-user-action=ignore

RUN pip install --root-user-action=ignore \
    -r requirements.txt \
    && pip cache purge

WORKDIR /root/workspace

CMD ["/bin/bash"]