main
黄海 1 year ago
parent 82feeac7d7
commit 5fd50d3b89

@ -6,4 +6,4 @@ config = ConfigUtil.getConfig()
server_address = config.get('comfyui', 'server_address') server_address = config.get('comfyui', 'server_address')
# 清理GPU缓存 # 清理GPU缓存
clearGPU(server_address) clear_cache(server_address)

@ -94,14 +94,14 @@ def getUse(server_address):
used_vram = vram_total - vram_free used_vram = vram_total - vram_free
used_vram_str = str(int((vram_total - vram_free) / 1024 / 1024 / 1024 + 0.5)) used_vram_str = str(int((vram_total - vram_free) / 1024 / 1024 / 1024 + 0.5))
used_lv = round(1.0 * (used_vram) / vram_total * 100, 2) used_lv = round(1.0 * (used_vram) / vram_total * 100, 2)
return "显存共" + vram_total_str + "GB,已使用" + used_vram_str + "GB,使用率:" + str(used_lv) + "% " return "显存共" + vram_total_str + "GB,已使用" + used_vram_str + "GB,使用率:" + str(used_lv) + "% "
# 如何清空Comfyui的gpu缓存 # 如何清空Comfyui的gpu缓存
# https://wailikeji.blog.csdn.net/article/details/140035515 # https://wailikeji.blog.csdn.net/article/details/140035515
# 清理GPU显存 # 清理GPU显存
def clearGPU(server_address): def clear_cache(server_address):
# 显示显卡使用率 # 显示显卡使用率
print('清理显存前:' + getUse(server_address)) print('清理显存前:' + getUse(server_address))
# 打开文件并读取内容 # 打开文件并读取内容

Loading…
Cancel
Save