You could try to run some code from our tutorials 30 to make sure PyTorch is working correctly.
A quick sanity check would be:
import torch
print(torch.__version__)
print(torch.randn(1, device='cuda'))

to make sure you can use the GPU properly.