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

pytorch是否使用了GPU的问题 #109

Open
Okiii-lh opened this issue May 29, 2020 · 3 comments
Open

pytorch是否使用了GPU的问题 #109

Okiii-lh opened this issue May 29, 2020 · 3 comments

Comments

@Okiii-lh
Copy link

作者你好，最近有在学习您的代码，使用的是pytorch版的神经网络，在跑代码的时候，发现GPU利用率几乎为0，我电脑是装好pytorch的，我现在在怀疑我电脑环境问题，但是又找不到问题，在其他issue中有看到是因为mcts过程过于复杂，导致cpu的计算速度跟不上GPU的速度，最终使得训练速度很慢，想问想我现在训练时GPU利用率很低是因为cpu计算速度跟不上吗还是我本地环境有问题，感谢您能花时间回复！
@dofish
Copy link

dofish commented Jun 14, 2020

训练的瓶颈是selfplay数据的获取过程，需要大量的cpu运算。

@TachikakaMin
Copy link

有个参数 "use_gpu=False"在 PolicyValueNet，可以确认一下是否打开。

@picasso250
Copy link

我用pytorch，打开了use_gpu，发现更慢了

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
4 participants
@picasso250 @TachikakaMin @Okiii-lh @dofish