suragnair / alpha-zero-general

A clean implementation based on AlphaZero for any game in any framework + tutorial + Othello/Gobang/TicTacToe/Connect4 and more
MIT License
3.74k stars 1.01k forks source link

self play is exceptionally slow on GPU #293

Open dberma15 opened 1 year ago

dberma15 commented 1 year ago

I'm using the code as is on Othello in tensoeflow and I'm finding it to be an order of magnitude slower on GPU compared to cpu. Is there a reason for this?

nullne commented 5 months ago

GPU isn't used when self play