跳到主要内容
首页/新闻中心/技术实践
技术实践PyTorch 项目

PyTorch 2.7 扩展 GPU 支持与编译缓存

官方发布说明涵盖 Blackwell 架构、可移植编译缓存,以及面向大模型的注意力推理更新。

以上日期为官方原文发布日期 · 本文为行业资讯简述

工程主题示意图,非活动现场照片

对接新一代计算架构

PyTorch 2.7 新增 Blackwell 架构的原型支持,提供 CUDA 12.8 预编译包,覆盖两类 Linux 架构。Triton 同步更新以支持编译执行。

让编译成果可复用

Mega Cache 作为测试版功能,允许保存并加载编译缓存,以减少后续运行的准备工作;注意力推理能力也有扩展。发布说明区分测试版与原型特性,便于逐项评估使用条件。

← 返回新闻中心