pytorch和cuda的版本适配问题

import torch

print(torch.__version__)#注意是双下划线


>>>import torch

>>>torch.version.cuda#输出一个 cuda 版本


torch.cuda.is_available()

你可能感兴趣的:(pytorch和cuda的版本适配问题)