今天 PyTorch 官方正式发布了 1.0 版本，不再是 Preview 了。我们来看一下主要更新了哪些内容吧。
更新日志
- 主要更新
- 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
- 性能优化
- 优化了官方文档
下载地址
为方便国内的小伙伴下载，这里列出常用的安装命令。
Linux, Conda, Python 3.6, CUDA 9
1 | conda install pytorch torchvision -c pytorch |
Linux, Conda, Python 3.6, CUDA 10
1 | conda install pytorch torchvision cuda100 -c pytorch |
Linux, pip, Python 3.6, CUDA 9
1 | pip3 install torch torchvision |
Linux, pip, Python 3.6, CUDA 10
1 2 | pip3 install http://download.pytorch.org/whl/cu100/torch-1.0.0-cp36-cp36m-linux_x86_64.whl pip3 install torchvision |
Linux, Conda, Python 2.7, CUDA 9
1 | conda install pytorch torchvision -c pytorch |
Linux, Conda, Python 2.7, CUDA 10
1 | conda install pytorch torchvision cuda100 -c pytorch |
Linux, pip, Python 2.7, CUDA 9
1 | pip install torch torchvision |
Linux, pip, Python 2.7, CUDA 10
1 2 3 4 5 | pip install http://download.pytorch.org/whl/cu100/torch-1.0.0-cp27-cp27mu-linux_x86_64.whl pip install torchvision # if the above command does not work, then you have python 2.7 UCS2, use this command pip install http://download.pytorch.org/whl/cu100/torch-1.0.0-cp27-cp27m-linux_x86_64.whl |
详细更新日志
具体更新日志大家可以移步官方查看：https://github.com/pytorch/pytorch/releases/tag/v1.0.0
本站微信群、QQ群(三群号 726282629)：
评论列表(1)