mirror of
https://github.com/hzwer/ECCV2022-RIFE.git
synced 2025-12-15 16:07:47 +01:00
23 lines
501 B
Docker
23 lines
501 B
Docker
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 |