PyTorch, Cuda, CuDNN 버전 확인하기

다음 코드로 확인 가능하다.

import torch

print("Torch version:{}".format(torch.__version__))
print("cuda version: {}".format(torch.version.cuda))
print("cudnn version:{}".format(torch.backends.cudnn.version()))