Dockerfile

FROM python:3.9

RUN useradd -m -u 1000 user
USER user
WORKDIR /home/user

COPY ./requirements.txt /home/user/requirements.txt

RUN pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
RUN pip install --no-cache-dir --upgrade -r /home/user/requirements.txt

COPY --chown=user . .
CMD chmod -R 777 /home/user

CMD ["python", "run.py"]