PyTorch 1.0 正式版发布(包含更新的安装命令)

PyTorch入门实战教程

今天 PyTorch 官方正式发布了 1.0 版本,不再是 Preview 了。我们来看一下主要更新了哪些内容吧。

更新日志

  • 主要更新
    • JIT
    • 全新的分布式训练包:torch.distributed
      • 旧的接口放到了 torch.distributed.deprecated
    • C++ 接口 API [尚不稳定]
    • Torch HUB:各种预训练模型集合
  • 重点更新
    • 众多新特性
      • 增加了对 N 维空的 tensor 的支持
      • 众多新的 tensor 操作
      • 新的分布函数
      • 对稀疏数据处理的 API 的改进
  • 修复了众多bug,比如 torch.nn.functional.softmin 在 0.4.1 里的公式是错的,torch.nn.MSELoss 有时候计算错误等等
  • 众多改进
  • 废弃的方法
    • 不再支持 C 接口
    • 不再支持 torch.utils.trainer
  • 性能优化
  • 优化了官方文档

下载地址

为方便国内的小伙伴下载,这里列出常用的安装命令。

Linux, Conda, Python 3.6, CUDA 9

Linux, Conda, Python 3.6, CUDA 10

Linux, pip, Python 3.6, CUDA 9

Linux, pip, Python 3.6, CUDA 10

Linux, Conda, Python 2.7, CUDA 9

Linux, Conda, Python 2.7, CUDA 10

Linux, pip, Python 2.7, CUDA 9

Linux, pip, Python 2.7, CUDA 10

详细更新日志

具体更新日志大家可以移步官方查看:https://github.com/pytorch/pytorch/releases/tag/v1.0.0

PyTorch入门实战教程
除特别注明外,本站所有文章均为 PyTorch 中文网原创,转载请注明出处:https://www.pytorchtutorial.com/pytorch-1-0/

Leave a Reply

Your email address will not be published. Required fields are marked *

返回顶部