search

junxiaosong / AlphaZero_Gomoku

An implementation of the AlphaZero algorithm for Gomoku (also called Gobang or Five in a Row)
MIT License
3.33k stars 970 forks source link

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

Open Okiii-lh opened 4 years ago

Okiii-lh commented 4 years ago

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

dofish commented 4 years ago

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

TachikakaMin commented 4 years ago

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

picasso250 commented 4 years ago

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