FROM cr-cn-beijing.volces.com/cloud-platform/colossalai:0.3.0

# install dependency
COPY ./requirements.txt /tmp/requirements.txt
RUN pip install -r /tmp/requirements.txt
