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

等待完成,過程比較慢

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM