Skip to content

[bug]: docker main-cpu image has gigabytes of nvidia/cuda stuff #7947

Open
@keturn

Description

@keturn

Is there an existing issue for this problem?

  • I have searched the existing issues

Operating system

Linux

GPU vendor

None (CPU)

Version number

5.10.1

What happened

I ran podman run --rm -it ghcr.io/invoke-ai/invokeai:main-cpu bash -i and it downloaded over 3 gigabytes of stuff.

Looking around in the venv's site-packages revealed that most of that is in nvidia libs and libtorch_cuda.so.

What you expected to happen

The docker images tagged cpu should be significantly smaller due to lack of CUDA dependencies. (…I guess a cpu-only venv is over 2 GB, so maybe not as much smaller as I was hoping, but still.)

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions