PyTorch logo


PyTorch is a machine learning library for Python, based on Torch

Status |
Overview    Platform    Social    System Requirements    Ratings    Developer    Written in    Initial Release    Repository    License    Categories    Alternatives   


PyTorch is an open-source machine learning library for Python, based on Torch, used for applications such as natural language processing. One can also reuse Python packages such as NumPy, SciPy and Cython to extend PyTorch when needed.

PyTorch provides two high-level features:

  • Tensor computation (like NumPy) with strong GPU acceleration
  • Deep neural networks built on a tape-based autodiff system - on GitHub repo

Automatic differentiation is done with a tape-based system at both a functional and neural network layer level. This functionality brings a high level of flexibility and speed as a deep learning framework and provides accelerated NumPy-like functionality.

PyTorch also includes standard defined neural network layers, deep learning optimizers, data loading utilities, and multi-gpu and multi-node support. Functions are executed immediately instead of enqueued in a static graph, improving ease of use and a sophisticated debugging experience. - NVIDIA - DEEP LEARNING FRAMEWORKS DOCUMENTATION

Blog   |  Documentation   |  Slack   |  FAQ   |  PyTorch Discuss Q&A   |  Wiki





System Requirements

1Linux distributions that use glibc >= v2.17
1macOS 10.10 (Yosemite) or above
1Windows 7 and greater; Windows 10 or greater recommended, Windows Server 2008 r2 and greater



InfoWorld 4.5 
based on professional's opinion


Adam Paszke(OD), Sam Gross(OD), Soumith Chintala(OD), Gregory Chanan(OD), Community

Written in

Python, C++, CUDA, C

Initial Release

24 August 2016



This page was last updated with commit: Following changes: (baef507)