简介:
PyTorch是目前最流行的深度学习框架之一,它是使用GPU和CPU优化的深度学习张量库。这是一个基于Python的科学计算包,主要针对:
1.替代Numpy有效的使用GPU
2. 一个深度学习研究平台,提供最大的灵活性和速度
PyTorch 作为库主要包含以下组件:
Torch:类似于 NumPy 的张量库,带有强大的 GPU 支持
torch.autograd:一个基于 tape 的自动微分库,支持 torch 中的所有的微分张量运算
torch.nn:一个专为最大灵活性而设计、与 autograd 深度整合的神经网络库
torch.multiprocessing:Python 多运算,但在运算中带有惊人的 torch 张量内存共享。这对数据加载和 Hogwild 训练很有帮助。
torch.utils:数据加载器、训练器以及其他便利的实用功能
torch.legacy(.nn/.optim):出于后向兼容性原因而从 torch 移植而来的旧代码
使用 PyTorch 的优势:
代替 NumPy 从而可以使用强大的 GPU
PyTorch 作为深度学习研究平台提供了最大的灵活性与速度
环境搭建
Window10 下安***r> 可以从这里下载安装包: https://pan.baidu.com/s/1dF6ayLr#list/path=/
Anaconda安***r> 先安装Anaconda,然后打开CMD执行下面命令:
conda install numpy mkl pyyaml cffi
pip install torch-0.3.0b0.591e73e-cp36-cp36m-win_amd64.whl
测试:
在python交互模式下,import torch
详细可以参考: http://blog.csdn.net/xiangxianghehe/article/details/78736482