如何通过 tf.config.experimental.set_memory_growth 解决显存被 TF 强行吃光的尴尬
在使用 TensorFlow (TF) 进行深度学习开发时,尤其是涉及到 GPU 资源管理时,许多开发者会遇到一个令人头疼的问题:TensorFlow 默认会在初始化时,预先分配几乎所有可用的 GPU 显存,即使模型非常小。这导致了显存资源...
在使用 TensorFlow (TF) 进行深度学习开发时,尤其是涉及到 GPU 资源管理时,许多开发者会遇到一个令人头疼的问题:TensorFlow 默认会在初始化时,预先分配几乎所有可用的 GPU 显存,即使模型非常小。这导致了显存资源...
如何解决 PyTorch 模型迁移至昇腾 NPU 时的算子性能瓶颈与精度漂移 在国产化替代的浪潮中,将深度学习模型从 CUDA 环境迁移到华为昇腾(Ascend)CANN 平台,绝非简单的 device=’cuda’...
如何解决昇腾 NPU 上频繁创建张量导致的内存碎片问题 在将模型从 CUDA 迁移到昇腾 NPU(Ascend)时,很多开发者会遇到一个诡异现象:通过 nvidia-smi 类似的工具观察,显存(HBM)占用并没满,但程序却频繁报出 Out...
引言 在训练大型深度学习模型时,显存(VRAM)往往是最大的瓶颈之一。TensorFlow 2.x 引入了强大的混合精度训练(Mixed Precision Training)功能,允许我们在不牺牲模型精度的情况下,大幅减少显存占用并提高训...
如何优化座舱多模态大模型的 KV Cache:解决长对话场景下的显存占用溢出难题 在智能座舱场景下,多模态大模型(VLM)需要实时处理语音、视觉(如驾驶员监控)和长文本上下文。由于座舱 SoC 的显存(如 Orin X 或骁龙 8295)通...
如何解决安卓多模型部署时的显存瓶颈:详解权重复用与Backbone共享技巧 在安卓移动端部署 AI 能力时,开发者常面临「内存焦虑」。当你需要在一个 App 中同时运行人脸检测、五官定位和属性识别时,如果每个模型都包含独立的骨干网络(Bac...
如何通过共享内存实现多进程模型权重共享:大幅降低 App 内存占用 在端侧推理或高并发 Web 服务场景中,为了提升吞吐量,我们常会启动多个进程并行处理推理请求。然而,如果每个进程都独立加载一份模型(例如一个 2GB 的 BERT 模型),...
如何利用 MUSA 运行时的显存超发机制解决大模型推理中的显存不足问题 在国产 GPU 适配的过程中,显存不足(OOM, Out of Memory)是运行大规模语言模型(LLM)时最常见的痛点。摩尔线程(Moore Threads)的 M...
在处理大规模深度学习部署时,经常遇到这样的问题:一块拥有 80GB 显存的高端 GPU(如 NVIDIA A100/H100)运行着许多只需要 5GB 或 10GB 显存的小模型。传统的 GPU 调度机制要么独占整块 GPU,造成资源浪费,...
别只知道 FlashAttention 的效果好,理解其背后的原理——解决显存带宽瓶颈——对于优化深度学习模型至关重要。标准 Self-Attention 机制在序列长度 $L$ 较大时,其性能瓶颈并非是计算量(FLOPs),而是显存的读写...