Pytorch 1.0升级到Pytorch 1.1.0

您所在的位置:网站首页 升级torch到指定版本 Pytorch 1.0升级到Pytorch 1.1.0

Pytorch 1.0升级到Pytorch 1.1.0

#Pytorch 1.0升级到Pytorch 1.1.0| 来源: 网络整理| 查看: 265

Pytorch 1.0Pytorch 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

等待完成,过程比较慢

 



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3