https://github.com/AUTOMATIC1111/stable-diffusion-webui/blob/master/modules/api/api.py

#秋叶整合包如何安装Python包
https://www.cnblogs.com/bossma/p/17593474.html

# 升级 pip
打开  D:\sd-webui-aki-v4.7\python\Scripts 然后在地址栏中输入cmd回车：

python -m pip install --upgrade pip

# 安装包 

pip  install pynvml 

# 修改的文件
D:\sd-webui-aki-v4.7\modules\api\api.py



# 添加的内容一
from pynvml import *
import torch
import gc

# 添加的内容二
self.add_api_route("/sdapi/v1/empty_cache", self.empty_cache, methods=["POST"])
self.add_api_route("/sdapi/v1/get_vram", self.get_vram, methods=["POST"])


# 添加的
def empty_cache(self):
    if torch.cuda.is_available():
        gc.collect()
        torch.cuda.empty_cache()
        torch.cuda.ipc_collect()
        gc.collect()
    return {"success":True,"message":"GPU is cleared!"}


def getVRam(self):
    nvmlInit()
    h = nvmlDeviceGetHandleByIndex(0)
    info = nvmlDeviceGetMemoryInfo(h)
    res=[]
    res.append(f'total: {round(info.total / 1024 / 1024 / 1024, 1)} GB')
    res.append(f'free : {round(info.free / 1024 / 1024 / 1024, 1)} GB')
    res.append(f'used : {round(info.used / 1024 / 1024 / 1024, 1)} GB')
    nvmlShutdown()
    return res