查看pytorch,cuda和cudnn版本号
查看pytorch,cuda和cudnn版本号import torchprint(torch.__version__)print(torch.version.cuda)print(torch.backends.cudnn.version())
·
查看pytorch,cuda和cudnn版本号
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())

更多推荐



所有评论(0)