2023-01-20 02:30:14 +01:00
|
|
|
FROM nvidia/cuda:11.7.1-runtime-ubuntu20.04
|
2022-10-05 17:15:09 +02:00
|
|
|
|
|
|
|
|
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 && \
|
2023-01-20 02:30:14 +01:00
|
|
|
pip3 install torch==1.13.1 xformers==0.0.16rc425
|
|
|
|
|
|
2022-11-01 02:09:43 +01:00
|
|
|
ARG version
|
|
|
|
|
|
|
|
|
|
RUN pip3 install lama-cleaner==$version
|
2022-10-05 17:15:09 +02:00
|
|
|
|
|
|
|
|
EXPOSE 8080
|
|
|
|
|
|
2022-11-01 02:09:43 +01:00
|
|
|
CMD ["bash"]
|
