Checking CUDA compatibility

1

Hi
How can I find whether pytorch has been built with CUDA/CuDNN support? Is there any log file about that?

2

You could use print(torch.__config__.show()) to see the shipped libraries or alternatively something like:

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

would also work.