import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())
挖个坑,因为我自己机器没有带cuda和cudnn,pytorch自带好像内置cudnn
知乎上有相关介绍
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())
挖个坑,因为我自己机器没有带cuda和cudnn,pytorch自带好像内置cudnn
知乎上有相关介绍