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