File size: 501 Bytes
c62dd62 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
FROM python:3.8-slim
# install deps
RUN apt-get update && apt-get -y install \
bash ffmpeg
# setup RIFE
WORKDIR /rife
COPY . .
RUN pip3 install -r requirements.txt
ADD docker/inference_img /usr/local/bin/inference_img
RUN chmod +x /usr/local/bin/inference_img
ADD docker/inference_video /usr/local/bin/inference_video
RUN chmod +x /usr/local/bin/inference_video
# add pre-trained models
COPY train_log /rife/train_log
WORKDIR /host
ENTRYPOINT ["/bin/bash"]
ENV NVIDIA_DRIVER_CAPABILITIES all |