File size: 410 Bytes
ed01507 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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"]
|