PyTorch是一个Python软件包,提供了两个高级功能: 具有强大GPU加速功能的Tensor计算(如NumPy) 基于基于磁带的autograd系统构建的深度神经网络 您可以在需要时重用自己喜欢的Python软件包(例如NumPy,SciPy和Cython)来扩展PyTorch。 系统 3.6 3.7 3.8 Linux CPU — Linux GPU — Windows CPU / GPU — — Linux(ppc64le)CPU — — Linux(ppc64le)GPU — — Linux(aarch64)CPU 另请参见 。 有关PyTor