2021-01-17 15:24:57 +01:00
|
|
|
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
|
|
|
|
|
|
2021-01-17 16:05:48 +01:00
|
|
|
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
|
2021-01-17 15:24:57 +01:00
|
|
|
|
|
|
|
|
# add pre-trained models
|
2021-01-17 16:07:54 +01:00
|
|
|
COPY train_log /rife/train_log
|
2021-01-17 15:24:57 +01:00
|
|
|
|
|
|
|
|
WORKDIR /host
|
2021-01-17 16:05:48 +01:00
|
|
|
ENTRYPOINT ["/bin/bash"]
|
2021-01-17 15:24:57 +01:00
|
|
|
|
|
|
|
|
ENV NVIDIA_DRIVER_CAPABILITIES all
|
