Dockerfile

FROM runpod/pytorch:2.2.1-py3.10-cuda12.1.1-devel-ubuntu22.04
WORKDIR /content
ENV PATH="/home/camenduru/.local/bin:${PATH}"
RUN adduser --disabled-password --gecos '' camenduru && \
    adduser camenduru sudo && \
    echo '%sudo ALL=(ALL) NOPASSWD:ALL' >> /etc/sudoers && \
    chown -R camenduru:camenduru /content && \
    chmod -R 777 /content && \
    chown -R camenduru:camenduru /home && \
    chmod -R 777 /home

RUN apt update -y && add-apt-repository -y ppa:git-core/ppa && apt update -y && apt install -y aria2 git git-lfs unzip ffmpeg

USER camenduru

RUN pip install -q opencv-python==4.9.0.80 imageio==2.34.1 imageio-ffmpeg==0.5.0 ffmpeg-python==0.2.0 av==12.1.0 runpod==1.6.2 \
    pillow==10.2.0 peft==0.11.1 einops==0.8.0 transformers==4.36.2 diffusers==0.27.2 accelerate==0.30.1 xformers==0.0.25 protobuf==3.20

COPY ./worker_runpod.py /content/worker_runpod.py
WORKDIR /content
CMD python worker_runpod.py