Install PyTorch GPU on Windows – A complete guide

PyTorch is an open-source machine learning framework originally from Meta Ai. If you want to study AI or work on AI projects, setting up a development environment with PyTorch on a local machine is essential for many projects. GPU (Graphics processing unit) is a specialized processor originally designed to process data simultaneously. It helps to speed up the computation of your deep-learning code. This is a complete guide to install PyTorch GPU on Windows. The installation involves many steps. Let’s get started.

Table of Content

  1. Download Nvidia graphics driver
  2. Install Visual Studio Community
  3. Install CUDA Toolkit 11.8.0
  4. Install cuDNN 8.6
  5. Install Anaconda3
  6. Create virtual environment for TyPorch
  7. Install PyTorch
  8. Run your project
  9. Install PyCharm


1. Download Nvidia graphics driver


If you already have a PC with GPU, you can skip this step. If you are going to get a new machine learning PC with GPU, please check CUDA enabled GPU cards before buying one. I suggest you buy a PC with a Nvidia graphic card having at least 12GB GDDR6 or GDDR6X. If you just got a new PC with Nvidia GPU, you go to Nvidia driver download and download the graphic card driver. After downloading, you run the exe file to install.

Nvidia driver downloads


2. Install Visual Studio Community

Note: if you already have installed the Nvidia CUDA toolkit and library, you can skip the next three steps.

To compile and support the Nvidia toolkit and library, you need a C++ compiler. You go to Visual Studio 2022. Go to Visual studio download page. Select community and click Free download. Run the installer. Select three elements to install: Python development, desktop c++, and visual studio extension development.

download visual studio 2022

Run the installer to finish the installation.


3. Install CUDA Toolkit 11.8.0

CUDA is a software layer that gives direct access to the Nvidia GPU’s virtual instruction set and parallel computational elements. According to Pytorch install guide, you need CUDA 11.8. Go to CUDA Toolkit archive at Nvidia, and select CUDA ToolKet 11.8.0. In the next window, select Windows, the version of your operation system, and the installer type, you can download and follow the default to install it.

download cuda toolkit

It is installed at C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8.


4. Install cuDNN 8.6

The Nvidia CUDA® Deep Neural Network Library (cuDNN) is a GPU-accelerated library of primitives for deep neural networks. Go to Nvidia’s cuDNN Archive, and download cuDNN v8.6 (not the latest).

download cuda toolkit

Extract the downloaded zip file and copy three subfolders, i.e. include, lib and bin.

cuDNN lib

Paste them to the CUDA install folder at C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\. New files are added to CUDA.

cuda folder

Open the Environment Variables window, and add the following new two paths:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\libnvvp

environment variables


5. Install Anaconda3

Conda is an open-source package and environment management system that runs on Windows, macOS, and Linux. Conda quickly installs, runs, and updates packages and their dependencies. To install, go to Anaconda site to download Anaconda installer, click the download button at the top, or go to the bottom to select the particular version for your PC.

download anaconda

After you run the installer, Anaconda3 is installed at C:\Users\yourusername\anaconda3. It also installs Python 3.10 there. Remember to add the following paths in “Environment Variables”:
C:\Users\yourusername\anaconda3
C:\Users\yourusername\anaconda3\Library\bin
C:\Users\yourusername\anaconda3\Scripts

environment variables

In Windows Start, you can find the Anaconda3 folder. If you cannot find it, search for “Anaconda prompt” in the window search bar at the bottom. Create a shortcut for the Anaconda prompt on your desktop. You will use it a lot.

Anaconda start folder

Run the Anaconda prompt, and a prompt window opens. It shows (base) C:\Users\yourusername>. (base) represents your current environment. “base” is a default env.

anaconda prompt

You can run conda or pip commands to see what packages have been installed.
$conda list
$pip list


6. Create virtual environment for TyPorch


A virtual environment is an isolated environment for Python projects. Each project can have its own dependencies, regardless of what dependencies every other project has. 

To run PyTorch, it is a good idea to create its own virtual environment. Open the Anaconda prompt window, and enter commands:
$conda create –name pytorch_env python=3.10
$conda activate pytorch_env

Pytorch virtual env

Now you are in the “pytorch_env” virtual env.


7. Install PyTorch


Go to PyTorch site to check the software compatibilities, as orange highlights in the following chart.

Pytorch software

Copy the command at the bottom and run it in your Anaconda prompt. This command installs PyTorch and other commonly used packages such as NumPy.

$pip3 install torch torchvision torchaudio –index-url https://download.pytorch.org/whl/cu118

To check whether Pytorch is installed correctly with GPU support, run Python in the Anaconda prompt.
$python
Then run
>>>import torch
>>>torch.cuda.get.device_name(0)
If it returns your GPU card name, you have successfully installed PyTorch with GPU support.

torch cuda


8. Run your project

You can run Python code in the Anaconda prompt window. For example, enter commands:
$python app.py

Sometimes, you need to install additional packages or modules. You can run pip or conda in Anaconda prompt. For example,
$pip install ipython
$conda install -c conda-forge ffmpeg


9. Install PyCharm

PyCharm is an integrated development environment used for programming in Python. If you want to edit and run your project in PyCharm, you need to install it first. Go to Download PyCharm at jebrains site. Download the Community version and run the installer.

After installation, open PyCharm and create a new project. In the Python interpreter, check “Previously configured interpreter”. In the drop-down, select the “pytorch_env” you just created.

You can also change the interpreter for existing projects. Go to file->settings->project. In Python Interpreter, check “Using existing environment”, click “Add interpreter”, and select “pytorch_env”.

torch cuda

Comments are closed