PyTorch 1.10の新機能「CUDA Graphs」のパフォーマンスを測定してみる
はじめに 10/21にPyTorch 1.10がリリースされ、今回も面白そうな機能が追加されました。個人的には楽しみにしていた「CUDA Graphs」のAPIのベータ版が追加されたということで早速試してみました。今回は … 続きを読む PyTorch 1.10の新機能「CUDA Graphs」のパフォーマンスを測定してみる