【Python】VRAM使用量を調べるデコレーターをClaude 3.5 Sonnetに書いてもらいました。

注意動作は問題なくします。Windowsタスクマネージャーが表示するVRAM使用量と一致していません。どちらが正確かはわかりません。Pythonスクリプト # pip install pynvml import functools import threading import time from pynvml import * import torch import gc def gpu_monitor(interval=1): def decorator(func): …