invoke-ai / InvokeAI

Invoke is a leading creative engine for Stable Diffusion models, empowering professionals, artists, and enthusiasts to generate and create visual media using the latest AI-driven technologies. The solution offers an industry leading WebUI, and serves as the foundation for multiple commercial products.
https://invoke-ai.github.io/InvokeAI/
Apache License 2.0
23.77k stars 2.44k forks source link

[bug]: invokeai docker v5.3.1 main-rocm still using cpu instead of rocm #7251

Open nmcbride opened 3 weeks ago

nmcbride commented 3 weeks ago

Is there an existing issue for this problem?

Operating system

Linux

GPU vendor

AMD (ROCm)

GPU model

RX 7700S, RX 7900 XTX

GPU VRAM

8, 24

Version number

5.3.1

Browser

Firefox

Python dependencies

No response

What happened

I was excited to see that https://github.com/invoke-ai/InvokeAI/issues/7146 was closed and merged in and all would be well in the rocm would however after updating my docker image and running with the new v5.3.1 the same issues persist and the container is using the cpu instead of rocm.

What you expected to happen

I expect rocm to be used.

How to reproduce the problem

No response

Additional context

No response

Discord username

No response

lonyelon commented 1 week ago

I can confirm this, the pytorch included is still CUDA, not ROCm.