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
评论0
最新资源