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.

Sign up for GitHub

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

Jump to bottom

Fix usage for multiple GPUs #325

Open
gcroci2 opened this issue Jan 20, 2023 · 1 comment
Open

Fix usage for multiple GPUs #325

gcroci2 opened this issue Jan 20, 2023 · 1 comment
Assignees
@gcroci2
Labels
Trainer trainer module related issues

Comments

@gcroci2
Copy link
Collaborator

gcroci2 commented Jan 20, 2023 •
edited
Loading

Everything works with 1 GPU and num_workers > 0, but if the number of GPUs is set to > 1 I get this error:

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cuda:0! (when checking argument for argument mat1 in method wrapper_addmm)

See these guides as starting point:
@gcroci2 gcroci2 added priority Solve this first Trainer trainer module related issues labels Jan 20, 2023
@github-actions
Copy link

This issue is stale because it has been open for 30 days with no activity. Remove stale label or comment or this will be closed in 7 days.

@github-actions github-actions bot added the stale issue not touched from too much time label Feb 20, 2023
@gcroci2 gcroci2 removed the priority Solve this first label Nov 3, 2023
@DaniBodor DaniBodor assigned DaniBodor and gcroci2 and unassigned DaniBodor Jul 9, 2024
@gcroci2 gcroci2 removed the stale issue not touched from too much time label Jul 9, 2024
@gcroci2 gcroci2 moved this to To do in Development Jul 12, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@gcroci2 gcroci2

Labels
Trainer trainer module related issues
Projects
Development
Status: To do
Milestone
No milestone
Development

No branches or pull requests
2 participants
@DaniBodor @gcroci2