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绘画。
Stable Diffusion常规绘图,总计用时229秒,平均每分钟可生成2.62张图像
测试显卡为NVIDIA GeForce RTX 4070 Super,测试选用模型为sd_xl_base_1.0_0.9vae.safetensors,测试中所选择的采样方法是Eular a,采样步数为50,图片宽度和高度为1024×1024,提示词相关系数为7,总批次数为10,单批数量为1,随机数种子为13(确保中可以生成相同的图像)。
经过TensorRT加速的Stable Diffusion绘图,总计用时118秒,平均每分钟可生成5.08张图像
可以看到在经过TensorRT加速后,用户能够获得2倍的AI生成速度,可以给予用户更加快速的AI生成体验。