信息发布→ 登录 注册 退出

终于!大家心心念念的PyTorch Windows官方支持来了

发布时间:2025-06-19

点击量:

机器之心整理

参与:机器之心编辑部

GitHub 发布地址:https://www./link/282da3b8ccda1203a2b8d6668f014365 PyTorch 官网:https://www./link/35df583a06ef378df8540f6db2d36534

PyTorch 从 0.3.1 版本升级到 0.4.0 版本,带来了一系列重要改进,其中最引人注目的就是对 Windows 系统的官方支持。在官方支持之前,Windows 用户需要依赖其他开发者提供的第三方 conda 包来安装 PyTorch。现在,用户可以直接从 PyTorch 首页获取通过 conda 或 pip 安装的命令行,或按照教程使用源文件进行安装。

除了 Windows 支持外,PyTorch 0.4.0 版本还优化了内存使用、增加了对多种概率分布的支持、改进了数据类型和张量处理。机器之心也尝试在 Windows 上安装了简单的 CPU 版 PyTorch,通过 pip 安装过程非常顺畅。

然而,使用 Conda 安装 PyTorch 时可能会遇到一些问题,例如 Conda 可能会提示找不到对应的包。

此次版本更新的目录如下:

  1. 重大核心变化包括:

    • Tensor/Variable 合并
    • 零维张量数据类型迁移指南
  2. 新增功能包括:

    • Tensors:全面支持高级索引
    • 快速傅立叶变换
    • 神经网络:计算时的存储权衡
    • bottleneck-识别代码中热点(hotspots)的工具
    • torch.distributions 24 个基础的概率分布
    • 增加 cdf、方差、信息熵、困惑度等
    • 分布式训练
    • 易于使用的 Launcher utility
    • NCCL2 后端
    • C++拓展
    • Windows 支持
    • ONNX 改进
    • RNN 支持
  3. 性能改进

  4. Bug 修复

蚂蚁金服举办首届金融科技领域算法类大赛——ATEC 蚂蚁开发者大赛人工智能大赛,点击「阅读原文」进入大赛官网了解比赛信息,比赛报名请使用PC端浏览器打开官网。

标签:# rnn  # 首页  # 第三方  # 升级到  # 可以直接  # 带来了  # 请使用  # 找不到  # 之心  # 傅立叶  # 官网  # bug  # https  # http  # pytorch  # 人工智能  # 算法  # github  # 数据类型  # conda  # pip  # 分布式  # 热点  # c++  # 工具  # 浏览器  # windows  # git  
在线客服
服务热线

服务热线

4008888355

微信咨询
二维码
返回顶部
×二维码

截屏,微信识别二维码

打开微信

微信号已复制,请打开微信添加咨询详情!