PyTorch 深度学习框架

用 Python 搭建和训练神经网络的深度学习框架,支持 GPU 加速

需要技术AI 整理,待开发者认领开发框架
评分人数不足呦探分评分人数不足大众分评分方法
这个页面由 AI 根据官网公开信息整理,可能有错。产品团队可以认领后修改。 认领这个产品

这个项目只有源代码,需要会编程才能用。

星标
10.5万
GitHub 下载量
34.7万
复刻
3.1万
主要语言
Python
许可证
Other
最近更新
复制安装指令(给 AI 编程工具)

粘贴到 Codex、Claude Code、Cursor、Trae 等 AI 编程工具里,回车就会开始安装。第一次复制前请先阅读风险提示。

这些文件由项目作者放在 GitHub 上,本站不开发、不托管这些软件,也不检测其代码;国内访问 GitHub 可能较慢。使用前请阅读 第三方软件风险提示和项目的许可证。 数据更新于 2026/11/01 报告问题

介绍

PyTorch 是一个以 Python 为核心的深度学习框架,提供可在 CPU 和 GPU 上运行的张量计算,以及基于自动求导的神经网络构建能力。它采用即时执行方式,代码逐行运行,网络结构可在运行中动态变化,调试比较直观。

它既可以当作能用 GPU 的 NumPy 替代品,也可以作为深度学习研究和开发平台。项目由社区驱动,能与 NumPy、SciPy 等库配合,并支持用 C/C++ 编写扩展。

能做什么

  • 进行切片、索引、线性代数、归约等张量运算,并用 GPU 加速
  • 借助自动求导和动态计算图构建、训练神经网络
  • 用 Python 或 C/C++ 扩展接口编写自定义网络层

运行需要:Python 环境,可用 pip 或 Conda 安装;GPU 可选,支持 NVIDIA CUDA、AMD ROCm 和 Intel GPU

这段介绍由 AI 根据项目的公开说明整理,可能有疏漏,以项目主页为准。

评测

0 条评测

还没有评测。用过的话,来写第一条吧。

写评测