AlphaZero_Gomoku icon indicating copy to clipboard operation
AlphaZero_Gomoku copied to clipboard

Published 20 hours ago • verified publisher icon junxiaosong

pytorch是否使用了GPU的问题

Open Okiii-lh opened this issue 4 years ago • 3 comments

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

Okiii-lh avatar May 29 '20 15:05 Okiii-lh

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

dofish avatar Jun 14 '20 09:06 dofish

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

TachikakaMin avatar Jun 24 '20 13:06 TachikakaMin

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

picasso250 avatar Oct 04 '20 08:10 picasso250