Install & get started with PyTorch

Here is a simple guideline to install PyTorch on Windows 10. Its equally easy to install in Linux , Mac.

I recommend having a conda environment. If you want to install a specific python version, check this page.

Name of the environment here: pytorch

conda create — name pytorch

conda activate pytorch to activate the conda env

Here is a official website of PyTorch. If you need to learn more about CUDA, check this blog.

My configurations of CUDA:

Run the command inside the activated environment

You if you already have a CUDA toolkit, you can check version by command nvcc — version

To check the installed verion of PyTorch: conda list torch

Take hints from following snip, to verify if CUDA is available and version of CUDA & Python:

Commands: python, import torch, torch.cuda.is_available(), torch.version.cuda

-Writer: Piyush Kulkarni (Data Scientist)