NSight DLProf Pytorch メモ

Automatic1111 WebUI で NSight が使えるか調査したが、バージョンが合わないので、WebUI のコード変更が必要になる。

NVIDIA プロファイラを用いた PYTORCH 学習最適化手法のご紹介

cuda version を調べる

Powershell で nvcc -V。

torch version

venv がある場合は Powershell で ./venv/Scripts/activate。

# python
import torch
print(torch.__version__)

Pytorch と DLProf のバージョン合わせ

DLProf Release Notes