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