Open In App

Install Pytorch on Windows

Improve
Improve
Like Article
Like
Save
Share
Report

In this article, we will learn how to install Pytorch on Windows. PyTorch is an open-source machine learning library based on the Torch library, used for applications such as computer vision and natural language processing, primarily developed by Facebook’s AI Research lab. It is free and open-source software released under the Modified BSD license.

Method 1: Using pip

Step 1: Check if python is already installed by entering the following command in the command prompt.

python –version

If this command runs successfully, and we are able to get a Python version then we are good to go or else install python by referring to How to install Python on Windows?

Step 2: Check if pip is already installed by entering the following command in the command prompt.

pip –version

If this command runs successfully, and we are able to get a pip version then we are good to go or else install pip by referring to this article Download and install pip Latest Version.

Step 3: Enter any one of the following commands (according to your system specifications) to install the latest stable release of Pytorch.

1. Compute Platform: CUDA 10.2, Nvidia Driver version should be >= 441.22

pip3 install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

2. Compute Platform: CUDA 11.1, Nvidia Driver version should be >= 456.38

pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

3. Compute Platform: CPU

 pip3 install torch==1.8.1+cpu torchvision==0.9.1+cpu torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

Step 4: Check if Pytorch is successfully installed by entering the following command in the command prompt.

pip3 show torch

Method 2: Using conda

Step 1: Check if conda is installed by entering the following command in Anaconda Prompt.

conda –version

If this command runs successfully, and we are able to get a conda version then we are good to go or else install Anaconda by referring to this article How to install Anaconda on windows?

Step 2: Open Anaconda Prompt in Administrator mode and enter any one of the following commands (according to your system specifications) to install the latest stable release of Pytorch.

1. Compute Platform: CUDA 10.2, Nvidia Driver version should be >= 441.22

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

2. Compute Platform: CUDA 11.1, Nvidia Driver version should be >= 456.38

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

3. Compute Platform: CPU

conda install pytorch torchvision torchaudio cpuonly -c pytorch

Step 3: Check if Pytorch is successfully installed by entering the following command in Anaconda prompt.

conda list -f pytorch


Last Updated : 06 Oct, 2021
Like Article
Save Article
Previous
Next
Share your thoughts in the comments
Similar Reads