How to check which cuda version my pytorch is using

[ptrblck]

You could check the linked CUDA version via print(torch.version.cuda).