PyTorch is a Python package that provides two high-level features:
- Tensor computation (like NumPy) with strong GPU acceleration
- Deep neural networks built on a tape-based autograd system
You can reuse your favorite Python packages such as NumPy, SciPy and Cython to extend PyTorch when needed.
| System | 3.6 | 3.7 | 3.8 |
|---|---|---|---|
| Linux CPU | — | ||
| Linux GPU | — | ||
| Windows CPU / GPU | — | — | |
| Linux (ppc64le) CPU | — | — | |
| Linux (ppc64le) GPU | — | — | |
| Linux (aarch64) CPU |
See also the ci.pytorch.org HUD.