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

problem #391

Closed

LinpengPan opened this issue Dec 24, 2019 · 2 comments

LinpengPan commented Dec 24, 2019

How to train this network with multiple GPUs?

jaysonph commented Jan 2, 2020 •

edited

Loading

@LinpengPan
I think you can refer to PyTorch website.
A short example taken from there:

net = torch.nn.DataParallel(model, device_ids=[0, 1, 2])
output = net(input_var) # input_var can be on any device, including CPU

Collaborator

Flova commented Aug 3, 2021

Duplicate of #520

Flova marked this as a duplicate of #520

Flova closed this as completed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment