FROM nvidia/cuda:11.7.1-runtime-ubuntu20.04

RUN apt-get update && apt-get install -y --no-install-recommends \
    software-properties-common \
    libsm6 libxext6 ffmpeg libfontconfig1 libxrender1 libgl1-mesa-glx \
    curl python3-pip

RUN pip3 install --upgrade pip && \
    pip3 install torch==1.13.1 xformers==0.0.16rc425

ARG version 

RUN pip3 install lama-cleaner==$version

EXPOSE 8080

CMD ["bash"]