文件名称:pytorch:具有强大GPU加速功能的Python中的张量和动态神经网络
文件大小:17.94MB
文件格式:ZIP
更新时间:2024-02-23 19:04:48
python machine-learning deep-learning neural-network gpu
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多处理,但跨进程的火炬张量具有神奇的内存共