GitHub用户Zagitta开发的开源工具gpu-unlock,把矛头对准了英伟达Ampere架构GeForce显卡上的一项限制:部分Tensor Core运算中的指令吞吐被锁住了。社区测试给出的数字是,RTX 3090的部分计算能力提升接近80%,但实际增益会随任务不同而变化。
开发者给出的解释是,采用FP32累加的FP16和BF16操作,在相关消费级显卡上的运行速率约为数据中心级显卡的一半。gpu-unlock的做法是修改硬件调度选择寄存器,把这道限制解除。
这套方案并非凭空出现,它沿用了此前用于解锁英伟达CMP矿卡性能的研究。具体实现上,一个Linux内核模块会在英伟达图形驱动程序加载之后执行,修改相关寄存器并重置GPU,再把控制权交还给未经修改的官方驱动。因此它不需要永久刷写显卡VBIOS,也不依赖修改过的固件或驱动。
社区兼容性报告里的数字更直观:
- RTX 3090的BF16矩阵乘法性能从77.5提升至129.6 TFLOPS
- RTX 3090的TF32性能从40.2提升至69.8 TFLOPS
- RTX 3090 Ti的BF16吞吐量从87.5提升至157.4 TFLOPS
不过,不同负载下的表现并不一致。一位RTX 3090用户测试本地大语言模型推理时,长上下文提示词处理速度提升16%,令牌生成速度基本没有变化。RTX 3090 Ti运行ComfyUI时,若干大量使用BF16的图像生成负载吞吐量提高约47%至59%。
那么游戏方面呢?英伟达的DLSS技术同样使用Tensor Core,这意味着某些AI辅助渲染操作可能会从中受益。但需要明确的是,目前还没有经过独立验证的游戏场景测试能证实该工具可以提升帧率。
兼容性上,该项目现阶段仅支持Linux,实验性的Windows移植版也只在特定系统上被证明可以运行。该解锁方式仍属实验方案,用在主力游戏电脑上存在风险。
换句话说,性能提升很大程度上取决于工作负载。对于手里已经有RTX 30系显卡、又愿意折腾Linux环境的用户,这个工具在部分负载上提供了一种提升计算性能的可能性;但它离“开箱即用”还有相当距离。
热门跟贴