Run this: https://github.com/QuantScientist/Deep-Learning-Boot-Camp/blob/master/day%2002%20PyTORCH%20and%20PyCUDA/PyTorch/01%20PyTorch%20GPU%20support%20test.ipynb
Show us the output,