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
另请参见 。
有关PyTorch的更多信息
从粒度上讲，PyTorch是一个包含以下组件的库：
零件
描述
Tensor库（如NumPy），具有强大的GPU支持
基于磁带的自动差分库，支持火炬中所有可差分的Tensor操作
编译堆栈（TorchScript），以从PyTorch代码创建可序列化和可优化的模型
与autograd深度集成的神经网络库，旨在提供最大的灵活性
Python多处理，但跨进程的火炬张量具有神奇的内存共
