PyTorch是一个Python软件包,提供两个高级功能: 具有强大GPU加速功能的Tensor计算(如NumPy) 在基于磁带的自动毕业系统上构建的深度神经网络 您可以在需要时重用自己喜欢的Python软件包(例如NumPy,SciPy和Cython)来扩展PyTorch。 系统 2.7 3.5 3.6 Linux CPU - Linux GPU - Windows CPU / GPU - - Linux(ppc64le)CPU - Linux(ppc64le)GPU - 另请参见ci.pytorch.org HUD 。 有关PyTorch的更多信息 从粒度上讲,PyTorch是一个由以下组件组成的库: 成分 描述 火炬 像NumPy这样的Tensor库,具有强大的GPU支持 火炬自动档 一个基于磁带的自动微分库,该库支持火炬中所有可微分的Tensor