AI制图——Stable Diffusion测试
除了拥有出色的游戏性能之外,GeForce RTX 40系列GPU还逐渐成为了创作者的法宝,其中AI就是工作效率的倍增器。
Stable Diffusion是一款热门的AI绘画生成工具,只需要输入一句话、或者特定的关键词,即可轻松生成一幅想要的图像。该软件基于AI计算,可广泛应用于平面和3D设计,通过借助扩散模型生成海量的图片,从而大幅缩短作图时间,显著提升效率。该软件由NVIDIA RTX GPU加速,显存容量越大,支持生成图像的分辨率越高,GPU级别越高,渲染速度越快。
AI绘图工具Stable Diffusion支持TensorRT,TensorRT是NVIDIA推出的一款高性能深度学习推理SDK,可优化推理性能、加速各种工作负载,并支持大型语言模型推理。
在下载并引用相关插件后,即可在Stable Diffusion中增添“TensorRT”标签,之后再经过一系列设定,就能通过TensorRT来加速AI绘画。
本次测试所使用的显卡为:
技嘉RTX 4080 SUPER GAMING OC 魔鹰 16G
NVIDIA GeForce RTX 3090 Ti
测试选用模型为sd_xl_base_1.0_0.9vae.safetensors
测试中所选择的采样方法是Eular a,采样步数为50,图片宽度和高度为1024×1024,提示词相关系数为7,总批次数为10,单批数量为1,随机数种子为13(确保中可以生成相同的图像)。
Stable Diffusion常规绘图(RTX 3090 Ti),总计用时173.3秒,平均每分钟可生成3.46张图像
Stable Diffusion常规绘图(RTX 4080 SUPER),总计用时136.8秒,平均每分钟可生成4.4张图像
经过TensorRT加速的Stable Diffusion绘图(RTX 4080 SUPER),总计用时63.5秒,平均每分钟可生成9.5张图像
通过对比测试可以发现,同样使用SD_XL进行Stable Diffusion常规绘图时,使用RTX 4080 SUPER显卡要比RTX 3090 Ti的性能强27%。而加入支持RTX 4080 SUPER的TensorRT加速后,RTX 4080 SUPER显卡的AI绘图效率大幅提升,对比RTX 3090 Ti高出175%,即便对比未开启TensorRT加速的RTX 4080 SUPER显卡本身,开启TensorRT加速后也拥有116%的效率提升,可以给予用户更加快速的AI生成体验。