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

@smth Thanks for the reply! Is installing from the binaries the same as following the instructions on the front page? Because I’ve done this but I get a CUDA library error when running

a = torch.randn(10).cuda()

Just to make sure, I updated Pytorch with my Anaconda installation via the following

conda install pytorch torchvision cuda80 -c soumith

Still I get the following error:

Thanks again for the support!