PyTorch 1.10の新機能「CUDA Graphs」のパフォーマンスを測定してみる

目次1 はじめに 2 CUDA Graphsとは?3 PyTorchでCUDA Graphsを使う3.1 CUDA GraphsのStream Captureの機能を使ってグラフを構築 3.2 構築したグラフを実行 3. … 続きを読む PyTorch 1.10の新機能「CUDA Graphs」のパフォーマンスを測定してみる