Spaces:
Running
Running
File size: 979 Bytes
475e066 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
# Must use a Cuda version 11+
# FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime
FROM pytorch/pytorch:1.13.0-cuda11.6-cudnn8-runtime
WORKDIR /
COPY ./parser /parser
COPY ./configs /configs
RUN mkdir /checkpoints
# Install git
RUN apt-get update && apt-get install -y --no-install-recommends \
git \
build-essential
# Install python packages
RUN pip3 install --upgrade pip
ADD requirements.txt requirements.txt
RUN pip3 install -r requirements.txt
# Install cv2 dependencies
RUN apt-get update
RUN apt-get install ffmpeg libsm6 libxext6 -y
# We add the banana boilerplate here
ADD server.py .
# Add your model weight files
# (in this case we have a python script)
ADD download.py .
# Add your custom app code, init() and inference()
ADD app.py .
ENV PYTHONPATH "${PYTHONPATH}:/parser:/upscaler"
#Alternative to using build args, you can put your token in the next line
#ENV HF_AUTH_TOKEN={token}
RUN python3 download.py
EXPOSE 8000
CMD python3 -u server.py
|