PyTorch 0.4.0版本变化记录

【2018-04-25 更新】PyTorch 0.4.0版本横空问世！！！

 

一、重大核心变化包括

• Tensor/Variable 合并

• 零维张量

• 数据类型

• 迁移指南

 

二、现添加的新特征包括

• Tensors：

• 全面支持高级索引
• 快速傅立叶变换

• 神经网络：

• 计算时的存储权衡
• bottleneck-识别代码中热点（hotspots）的工具

• torch.distributions

• 24 个基础的概率分布
• 增加cdf、方差、信息熵、困惑度等

• 分布式训练

• 易于使用的 Launcher utility
• NCCL2 后端

• C++拓展

• Windows 支持

• ONNX 改进

• RNN 支持

 

三、性能改进

 

四、Bug 修复

 

五、torchvision的一些变化

 

1.torchvision.transform中函数torchvision.transforms.Scale(*args, **kwargs)即将被函数torchvision.transforms.Resize(size, interpolation=2)代替。（参考官方文档：点击打开链接）

2.torchvision.transform中函数torchvision.transforms.RandomSizedCrop(*args, **kwargs)即将被函数torchvision.transforms.RandomResizedCrop(size, scale=(0.08, 1.0), ratio=(0.75, 1.3333333333333333), interpolation=2)代替。（参考官方文档：点击打开链接）

 

六、安装方式——Windows安装

【方法一】pip直接安装。官网（点击打开链接）给出的安装步骤如下图所示（根据CUDA版本以及Python版本选择）。

 

【方法二】利用源文件安装。GitHub说明中给出的方法（点击打开链接）。注意必须安装有VS2017！（官方还给出了NVTX，但是我并没有找到相关的下载。。。）

（1）首先安装必须的包：

[plain] view plain copy

1. conda install numpy pyyaml mkl mkl-include setuptools cmake cffi typing  

 

（2）然后从GitHub上克隆：

 

[plain] view plain copy

1. git clone --recursive https://github.com/pytorch/pytorch  

 

（3）最后在windows命令提示行或者powershell中执行：

 

[plain] view plain copy

1. set "VS150COMNTOOLS=C:\Program Files (x86)\Microsoft Visual Studio\2017\Enterprise\VC\Auxiliary\Build"  
2. set CMAKE_GENERATOR=Visual Studio 15 2017 Win64  
3. set DISTUTILS_USE_SDK=1  
4. REM The following line is needed for Python 2.7, but the support for it is very experimental.  
5. set MSSdk=1  
6.   
7. call "%VS150COMNTOOLS%\vcvarsall.bat" x64 -vcvars_ver=14.11  
8. python setup.py install