PyTorch是一个开源的Python机器学习库,是基于Torch开发的用于自然语言处理等应用的开源库。PyTorch可以看作是加入了GPU支持的numpy,同时也可以看成是一个拥有自动求导功能的深度神经网络。PyTorch的前身是Torch,其底层和Torch框架是一样的,但是使用Python重写了很多内容,相对来讲使用更加灵活、内容更加丰富。它是由Torch7团队开发。
使用PyTorch搭建并训练深度学习模型需要导入需要的库和模块。步骤、流程和2.3.3节要讲的百度PaddlePaddle基本一致。