Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Hardware requirements? #1

Open
9876691 opened this issue Jul 29, 2023 · 4 comments
Open

Hardware requirements? #1

9876691 opened this issue Jul 29, 2023 · 4 comments

Comments

@9876691
Copy link

9876691 commented Jul 29, 2023

Could we list the hardware requirements or the hardware it's been tested on at the moment?

So I can see it's using the WGPU backend do you know roughly how much VRAM is needed?

Thanks.

@overheat
Copy link

I have same question. And does it works on Apple Silicon, M1/M2?

@nickgarfield
Copy link

nickgarfield commented Jul 30, 2023

I'm currently getting No CPU device found on M2 when trying to run the convert binary. Issue seems related to the latest commit (c512786). Reverting to 427911f worked for me.

Update:
Nevermind, I ran into a buffer size issue:

thread 'main' panicked at 'wgpu error: Validation Error

Caused by:
    In Device::create_buffer
      note: label = `Buffer Src`
    Buffer size 524288000 is greater than the maximum buffer size (268435456)

', /Users/nickgarfield/.cargo/registry/src/index.crates.io-6f17d22bba15001f/wgpu-0.17.0/src/backend/direct.rs:3056:5
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

@Gadersd
Copy link
Owner

Gadersd commented Jul 30, 2023

I'm currently getting No CPU device found on M2 when trying to run the convert binary.

I'll check with the burn team about this.

Once the burn caching functions are public I'll try benchmarking it.

@Gadersd
Copy link
Owner

Gadersd commented Aug 7, 2023

burn-wgpu doesn't use the full device memory so overflows happen with some large models but hopefully in a few days I'll have it working.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants