mirror of
https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg
synced 2025-04-19 12:19:49 +08:00
15 lines
354 B
Plaintext
15 lines
354 B
Plaintext
FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
|
|
|
|
WORKDIR /rembg
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends python3-pip python-is-python3 curl && apt-get clean && rm -rf /var/lib/apt/lists/*
|
|
|
|
COPY . .
|
|
|
|
RUN python -m pip install ".[gpu,cli]" --break-system-packages
|
|
RUN rembg d u2net
|
|
|
|
EXPOSE 7000
|
|
ENTRYPOINT ["rembg"]
|
|
CMD ["--help"]
|