2019-03-01から1ヶ月間の記事一覧

PyTorchのMulti-GPUでメモリ使用量の偏りと改善策

よくあるGPUメモリの偏り PyTorchでマルチGPUをすると, こんな感じで 1つのGPUのメモリだけ他に比べて多い (2倍以上)場合って OOMでバッチサイズが増やせずに悲しいですよね。 解決法 以下の方法で,8 GPUでOOMしないバッチサイズが16->32になりました…