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

Low GPU utility #6

Open
TianrongChen opened this issue Sep 22, 2021 · 3 comments
Open

Low GPU utility #6

TianrongChen opened this issue Sep 22, 2021 · 3 comments

Comments

@TianrongChen
Copy link

Hi Song Yang! Thank you for your code!
I am running your Network in a local (single GPU: RTX TITAN) machine, but the GPU utility is only around 30% during sampling. Did you notice the same issue?
Best Wishes,
Tianrong

@yang-song
Copy link
Member

I didn't notice this issue before. Sometimes the CPU can also be a bottleneck. Maybe it will be better when using a more powerful CPU?

@TianrongChen
Copy link
Author

Thank you for your reply! I figured it out XD
I think in my own code,
torch.backends.cudnn.deterministic = True
caused this problem.
Just turn this off, the problem will be solved.

@yang-song
Copy link
Member

Great to know you figured it out! Yes you're right. Setting torch.backends.cudnn.deterministic = True ensures reproducibility but can greatly reduce the efficiency of GPUs.

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

2 participants