TencentARC/GFPGAN
1
0
Fork 0
mirror of https://github.com/TencentARC/GFPGAN.git synced 2026-04-24 06:41:32 +00:00
Code Issues 476 Packages Projects Releases 8 Wiki Activity

How to specify the device #423

New Issue
Open
opened 2026-01-29 21:47:38 +00:00 by claunia · 2 comments
claunia commented 2026-01-29 21:47:38 +00:00
Owner
Copy Link

Originally created by @Crestina2001 on GitHub (Dec 25, 2023).

Hi, I am using the model for inference. I have several GPUs, and I would like to split the images for different GPUs to infer. How to move the restorer to different GPUs? restorer.to(device) does not work.

Originally created by @Crestina2001 on GitHub (Dec 25, 2023). Hi, I am using the model for inference. I have several GPUs, and I would like to split the images for different GPUs to infer. How to move the restorer to different GPUs? restorer.to(device) does not work.
claunia commented 2026-01-29 21:47:38 +00:00
Author
Owner
Copy Link

@Silas1704 commented on GitHub (Jan 23, 2024):

@Crestina2001 In PyTorch try using torch.nn.DataParallel or torch.nn.parallel.DistributedDataParallel .when using DataParallel, the model will be replicated across all available GPUs, and each GPU will process a portion of the input data Wrap the model with DataParallel:
restorer = nn.DataParallel(restorer).
Then concatenate the results

@Silas1704 commented on GitHub (Jan 23, 2024): @Crestina2001 In PyTorch try using torch.nn.DataParallel or torch.nn.parallel.DistributedDataParallel .when using DataParallel, the model will be replicated across all available GPUs, and each GPU will process a portion of the input data Wrap the model with DataParallel: restorer = nn.DataParallel(restorer). Then concatenate the results
claunia commented 2026-01-29 21:47:38 +00:00
Author
Owner
Copy Link

@jdc4429 commented on GitHub (Jul 20, 2024):

what about specifing a GPU other then 0? I am trying to get it to use gpu1 but my attempts have failed.

@jdc4429 commented on GitHub (Jul 20, 2024): what about specifing a GPU other then 0? I am trying to get it to use gpu1 but my attempts have failed.
Sign in to join this conversation.
No Branch/Tag Specified
Branches Tags
Labels
Clear labels
pull-request
Mirrored from GitHub Pull Request
No Label
Milestone
No items
No Milestone
Projects
Clear projects
No project
Assignees
Clear assignees
claunia
No Assignees
1 Participants
Notifications
Due Date
No due date set.
Dependencies

No dependencies set.

Reference: TencentARC/GFPGAN#423
Delete Branch "%!s()"

Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?