why docker image size is very big about rembg

[danielgatis]

because of cuda's dependencies

FROM nvidia/cuda            <---------------- here!

RUN apt-get update && apt-get install -y --no-install-recommends python3 python3-pip python3-dev llvm llvm-dev
RUN pip3 install rembg

ENTRYPOINT ["rembg"]
CMD []

[lekoOwO]

I've installed it into a container with no cuda deps. (My server has no GPU installed)

Do you need any help with it?

[ooing]

@lekoOwO can you show your Dockerfile

[a-stankevich]

I noticed that too, using image from danielgatis/rembg:latest.

At the same time ibebbs/u2net-http image is 1.82GB total.

[lekoOwO]

@lekoOwO can you show your Dockerfile

I run on a LXC Container, so I have no Dockerfile to share.
Here's the things I've installed on my Ubuntu container with no GPU:

apt install python3.8 python3.8-dev python3.8-pip python3.8-distutils llvm-10 
python3.8 -m pip install torch==1.7.0 torchvision==0.8.1 -f https://download.pytorch.org/whl/cpu/torch_stable.html
python3.8 -m pip install rembg

I think this might somewhat work. Not tested.

FROM ubuntu:latest
RUN apt-get update && apt-get install -y --no-install-recommends python3.8-dev python3.8-pip python3.8-distutils llvm-10 && pip3 install torch==1.7.0 torchvision==0.8.1 -f https://download.pytorch.org/whl/cpu/torch_stable.html
RUN pip3 install rembg

ENTRYPOINT ["rembg"]
CMD []