Training Trains Internship
Student login
← PyTorch learning path
PyTorch · Module 1 of 10

Getting started with PyTorch

PyTorch is a flexible, Pythonic deep learning framework.

📖 Key ideas

  1. PyTorch was developed by Meta and is now under the PyTorch Foundation.
  2. Popular in research and industry; most new models ship in PyTorch.
  3. Install from pytorch.org with the right CUDA version.
  4. Dynamic computation graphs feel like normal Python.
  5. Ecosystem: torchvision, torchaudio, Hugging Face, Lightning.
💡 Real-life example: PyTorch is like LEGO with instructions you can change while building.
PyTorchDeep learning framework
CUDANVIDIA GPU computing platform
Dynamic graphGraph built as code runs
torchvisionVision datasets and models