Ben Gorman

Ben Gorman

Life's a garden. Dig it.

Skip to the problems ->

PyTorch is an open source machine learning framework that accelerates the path from research prototyping to production deployment.

See the official PyTorch site for details and documentation.

How to install pytorch

For vanilla CPU-enabled PyTorch on Mac or Windows,

pip3 install torch torchvision torchaudio

If you're on Linux or you want to enable GPU, see the official installation guide.

Google Colab

To get up and running quickly, consider using Google Colab - a free ipython notebook that runs in the browser. PyTorch is installed on Google Colab by default 😃.

How to import PyTorch

By convention,

import torch