Можно ли заставить узлы и локоны MCTS работать только на GPU с PyTorch? - PullRequest
0 голосов
/ 15 февраля 2020

Я смотрел некоторые обсуждения о MCTS и GPU. Говорят, что использование GPU не дает никаких преимуществ, поскольку в нем не много матричных умножителей. Но у него есть недостаток, связанный с использованием ЦП, поскольку передача данных между устройствами действительно требует времени. Здесь я имею в виду узлы и дерево должны быть на GPU. Затем они могут обрабатывать данные на GPU, не копируя данные с CPU. Если я просто создам узел класса и дерево, они позволят своим методам работать на CPU. Поэтому мне интересно, могу ли я переместить поисковую часть в GPU. Есть ли пример?

...