Featured Post
我们很高兴地宣布发布 PyTorch® 2.5（release note）！此版本引入了一个新的 CuDNN 后端用于 SDPA，默认情况下为使用 H100 或更新 GPU 的 SDPA 用户提...
Read MoreJuly 24, 2024
我们很高兴地宣布 PyTorch® 2.4 的发布（release note）！ PyTorch 2.4 为 torch.compile 添加了对 Python（3.12）最新版本 的支持。AOTInductor freezing 通过允许序列化 MKLDNN 权重， 为运行 AOTInductor 的开发者提供了更多基于性能的优化。此外，引入了一个新的使用 libuv 的默认 TCPStore 服务器后端，能显著减少运行大规模作业时的初始化时间。 最后，新的 Python 自定义算子 API 使得将自定义内核集成到 PyTorch 中变得比以前更容易，特别是对于 torch.compile。
选择您的偏好并运行安装命令。 Stable 代表当前经过测试和支持的 PyTorch 版本，适合大多数用户。如果您想要最新的、尚未完全测试和支持的每日构建版本， 可以选择 Preview。在安装之前，请确保您已经满足下面的先决条件（例如，numpy），这取决于您的包管理器。我们推荐使用 Anaconda 作为包管理器，因为它会安装所有依赖项。 您也可以安装以前版本的 PyTorch。请注意，LibTorch 仅适用于 C++。
注意: 最新PyTorch版本需配套使用Python 3.8或更高版本
To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. As the current maintainers of this site, Facebook’s Cookies Policy applies. Learn more, including about available controls: Cookies Policy.
