在DGX Spark上用TensorRT Edge-LLM运行Nemotron-3.5 Lightning
NVIDIA技术博客教程介绍了如何在64GB DGX Spark上使用TensorRT Edge-LLM运行纯文本模型Nemotron-3.5 Lightning。trtedge命令行工具可下载模型并构建推理引擎,工作流涵盖终端聊天测试以及通过本地兼容OpenAI的API提供模型服务。教程还演示了可选的DSpark推测解码,该方法使用草稿模型提出令牌,再由主模型一次性验证。NVIDIA称,在SpeedBench 1K、SpeedBench 2K和HumanEval提示词工作负载上,启用DSpark后的平均生成速度更高。这些测试运行于搭载DGX OS 7.6.0的系统,采用NVFP4、批大小为1、单个并发流式请求,并限制输出为128个令牌。
该教程为嵌入式开发者提供了一套在DGX Spark上部署、测试和提供文本模型服务的具体流程,其中包含可配置的推测解码选项。教程还公布了NVIDIA在明确测试条件下测得的生成速度和内存使用数据,可用于评估该工作流。
AI is expanding beyond the data center as developers build a new generation of agents to act at the edge: monitoring driver attention and cabin conditions, helping robots understand their surroundings and plan tasks, and coordinating fleets of machines on production lines. These physical AI applications rely on models that perceive, reason, and act within tight memory, power, and latency budgets. But getting a model ready to run can involve substantial setup.
NVIDIA DGX Spark, now available in a 64GB configuration, gives embedded developers a platform to prototype, fine-tune, and test models and agents. NVIDIA TensorRT Edge-LLM provides a lightweight C++ inference runtime designed for embedded workloads, bringing the same inference framework to development on Spark and deployment on platforms such as NVIDIA Jetson Thor, NVIDIA DRIVE AGX Thor, and NVIDIA IGX Thor.
You can get started in just a few commands with the trtedge command-line interface (CLI). It downloads your selected model and builds TensorRT engines optimized for inference on Spark.
In this tutorial, you’ll run NVIDIA Nemotron-3.5 Lightning, a text-only model, on Spark using TensorRT Edge-LLM. You’ll test prompts in your terminal and serve the model through an API for application prototyping. You’ll also enable DSpark speculative decoding to accelerate generation.
#NVIDIA#TensorRT Edge-LLM#DSpark#DGX Spark 64GB#Nemotron-3.5 Lightning#trtedge#TensorRT#Jetson Thor#DRIVE AGX Thor#IGX Thor