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
等待完成,过程比较慢