pytorch/pytorch:GPU 加速张量与动态神经网络
来源:GitHub — 2026-10-11 · ⭐ 104,083 stars · 今日 +81 · Python
📋 概述
PyTorch 提供两块核心能力:像 NumPy 那样但具备强 GPU 加速的张量计算,以及建立在磁带式自动求导之上的深度神经网络。它坚持 Python 优先与命令式编程体验,调试方式与写普通 Python 一致,同时可以继续复用 NumPy、SciPy、Cython 等既有包来扩展。README 覆盖二进制与源码两条安装路径,并分别说明 NVIDIA CUDA、AMD ROCm 与 Intel GPU 的支持前提。以 Python 实现,累计约 10.4 万星、单日新增 81 星。
🔑 核心要点
- 提供 GPU 加速的张量计算,以及基于磁带式自动求导的神经网络两套能力。
- 坚持 Python 优先与命令式体验,调试方式与写普通 Python 一致。
- 可以继续复用 NumPy、SciPy、Cython 等既有包来扩展功能。
- 二进制与源码两条安装路径,分别说明 CUDA、ROCm 与 Intel GPU 支持,累计约 10.4 万星。
💡 金句
让研究者少写胶水代码,比多给几个算子更能加快实验的节奏。
👍 0
👎 0
← 返回 GitHub Trending 首页