Newbie question: What are the prerequisites for running PyTorch with GPU?

3

if you want to use pytorch with an NVIDIA GPU, all you need to do is install pytorch binaries and start using it. We ship with everything in-built (pytorch binaries include CUDA, CuDNN, NCCL, MKL, etc.).

So it’s a one-touch install:

  • Install pytorch from binaries ( http://pytorch.org has instructions)
  • Try a simple program with CUDA:
import torch
a = torch.randn(10).cuda()
print(a)
print(a + 2)
12 Likes