How to setup NVIDIA GPU for PyTorch on Windows 10/11

Channel:

SL7 Tech

Subscribers:

8,890

Published on April 10, 2023 4:37:40 PM ● Video Link: https://www.youtube.com/watch?v=r7Am-ZGMef8

Category:

Tutorial

Duration: 13:14

54,175 views

843

Go to https://strms.net/factor75_sl7tech and use code FACTORSE35503 for my special Factor75 discount and to support my channel! #ad
This is a tutorial video on how to setup NVIDIA graphics for using it with deep learning library PyTorch.

Links used:
Visual Studio 2019: https://visualstudio.microsoft.com/vs/older-downloads/
NVIDIA CUDA: https://developer.nvidia.com/cuda-toolkit-archive
NVIDIA CUDNN: https://developer.nvidia.com/rdp/cudnn-archive
PyTorch : https://pytorch.org/get-started/locally/

Code to test:

import torch
print(torch.cuda.device_count())
print(torch.cuda.get_device_name(0))