SSSxCCC / AlphaZero-In-Unity

Several board game AI, which are trained based on AlphaZero, used in a Unity game.
MIT License
39 stars 15 forks source link

Multi gpu model, low GPU utilization #1

Closed tensorgpu closed 5 years ago

tensorgpu commented 5 years ago

Training using GPU only utilize 1% of my GPU, anybody able to enhance it?

Maybe using keras.utils, multi_gpu_model?

tensorgpu commented 5 years ago

使用GPU的训练仅使用我的1%的GPU,谁可以增强它?

也许使用keras.utils,multi_gpu_model?

SSSxCCC commented 5 years ago

Most of time is spent in MCTS algorithm while training, so gpu won't help much unless you use a deeper neural network. 训练时绝大多时间花费在蒙特卡洛树搜索上面,所以用gpu加速效果不明显,除非用更深的神经网络。