File size: 386 Bytes
20bc8b5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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"]