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 1.2k has instructions)
Try a simple program with CUDA:

import torch
a = torch.randn(10).cuda()
print(a)
print(a + 2)