SapnilPatel/Setup-NVIDIA-GPU-for-Deep-Learning

Setup-NVIDIA-GPU-for-Deep-Learning

Step 1: NVIDIA Video Driver

You should install the latest version of your GPUs driver. You can download drivers here:

• NVIDIA GPU Drive Download

Step 2: Visual Studio C++

You will need Visual Studio, with C++ installed. By default, C++ is not installed with Visual Studio, so make sure you select all of the C++ options.

• Visual Studio Community Edition

Step 3: Anaconda/Miniconda

You will need anaconda to install all deep learning packages

• Download Anaconda

Step 4: CUDA Toolkit

• Download CUDA Toolkit

Step 5: cuDNN

• Download cuDNN

Step 6: Install PyTorch

• Install PyTorch

Finally run the following script to test your GPU

import torch

print("Number of GPU: ", torch.cuda.device_count())
print("GPU Name: ", torch.cuda.get_device_name())


device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
print('Using device:', device)