Pytorch 1.0升级到Pytorch 1.1.0

Pytorch 1.0

Pytorch 1.0于2018-12-8发布，详见https://github.com/pytorch/pytorch/releases/tag/v1.0.0

主要更新

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

性能优化

升级

本文中pytorch由0.4.0升级到1.0.1，再升级至1.1.0(也可以直接升级至1.1.0)

1. 首先用如下命令查看升级前的pytorch版本，返回版本号是0.4.0

python -c "import torch; print(torch.__version__)"

2. 输入如下命令 conda install pytorch torchvision -c pytorch进行升级

conda install pytorch torchvision -c pytorch

等待完成，过程比较慢