IOPaint/docker/GPUDockerfile

20 lines
587 B
Plaintext
Raw Normal View History

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
2023-11-01 02:51:26 +01:00
RUN pip3 install --upgrade pip
2023-11-01 14:14:29 +01:00
RUN pip3 install torch==2.1.0 torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu118
RUN pip3 install xformers==0.0.22.post4 --index-url https://download.pytorch.org/whl/cu118
2023-01-20 02:30:14 +01:00
2022-11-01 02:09:43 +01:00
ARG version
RUN pip3 install lama-cleaner==$version
2023-11-01 14:14:29 +01:00
RUN lama-cleaner --install-plugins-package
2022-10-05 17:15:09 +02:00
EXPOSE 8080
2022-11-01 02:09:43 +01:00
CMD ["bash"]