Chadrick Blog

check cuda version build to torch package and find cudnn version used in torch

use the following python snippet to check cuda version the torch package was built against

import torch
print(torch.version.cuda)

use the following python snippet to check cudnn version used by torch

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