Some GPU shortcuts

Here’s the table of contents:

1. Check if GPU is being used in PyTorch
2. Check which CuDNN version is used

Check if GPU is being used in PyTorch

print(torch.cuda.is_available())

See!

Check which CuDNN version is used

print(torch.backends.cudnn.version())

See!