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

Change the number of gpus #8

Closed
wcxchunxin opened this issue Apr 30, 2019 · 4 comments
Closed

Change the number of gpus #8

wcxchunxin opened this issue Apr 30, 2019 · 4 comments

Comments

@wcxchunxin
Copy link

@akanimax.Hi,
Nice work,I've implemented your code.It helped me a lot. But,I want to change the number of gpus.I own two Tesla graphics CARDS.Would you please tell me how to modify it! Thanks!

@BlindElephants
Copy link

If you're using cuda, the code provided in this repo should automatically be enabling parallel processing on available GPU hardware:

See here:

    if device == th.device("cuda"):
        self.gen = DataParallel(self.gen)
        self.dis = Discriminator(depth, latent_size, use_eql=use_eql, gpu_parallelize=True).to(device)

@wcxchunxin
Copy link
Author

wcxchunxin commented May 27, 2019 via email

@akanimax
Copy link
Owner

@wcxchunxin, Glad to know that your problem is solved.

@BlindElephants, Thanks a lot for helping @wcxchunxin out. You do seem to have spent a lot of time in this code. 😄. Thanks again.

Best regards,
@akanimax

@wcxchunxin
Copy link
Author

wcxchunxin commented May 27, 2019 via email

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

3 participants