Skip to content

Add GPU selection#405

Open
ZeusAFK wants to merge 1 commit into
lllyasviel:mainfrom
ZeusAFK:main
Open

Add GPU selection#405
ZeusAFK wants to merge 1 commit into
lllyasviel:mainfrom
ZeusAFK:main

Conversation

@ZeusAFK

@ZeusAFK ZeusAFK commented Apr 29, 2025

Copy link
Copy Markdown

image

This update improves the initialization process by:

  • Detecting and listing all available GPUs on the system.
  • Prompting the user to select a preferred GPU before any model loading begins.

@tugalsan

tugalsan commented May 5, 2025

Copy link
Copy Markdown

I have LG14, and framepack uses integrated GPU in Turbo Mode :...(

@oskapt

oskapt commented May 8, 2025

Copy link
Copy Markdown

you can set the env var CUDA_VISIBLE_DEVICES to the GPU you want framepack to use, and that's the one that it will see:

$ export CUDA_VISIBLE_DEVICES=1  # for GPU #1
$ python3 ./demo_gradio_f1.py

or all-in-one: CUDA_VISIBLE_DEVICES=1 python3 ./demo_gradio_f1.py

@ZeusAFK

ZeusAFK commented May 9, 2025

Copy link
Copy Markdown
Author

you can set the env var CUDA_VISIBLE_DEVICES to the GPU you want framepack to use, and that's the one that it will see:

$ export CUDA_VISIBLE_DEVICES=1  # for GPU #1
$ python3 ./demo_gradio_f1.py

or all-in-one: CUDA_VISIBLE_DEVICES=1 python3 ./demo_gradio_f1.py

No way, really? awesome, thanks!

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

Successfully merging this pull request may close these issues.

3 participants