2020-12-13から1日間の記事一覧

nn.DataParallelを使えばデータをdeviceに送らなくていい

この投稿はrioyokotalab Advent Calendar 2020 13日目の投稿です。 adventar.org nn.DataParallel PyTorchでマルチGPUを使った機械学習を行いたい場合、 nn.DataParalllel DistributedDataParallel の二種類の方法があります。 DistributedDataParallelはマ…