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())

0 Comments

Leave a Reply

Your email address will not be published.