import torch
print("Pytorch version:")
print(torch.__version__)
print("CUDA Version: ")
print(torch.version.cuda)
print("cuDNN version is :")
print(torch.backends.cudnn.version())
查看pytorch+cuda+cudnn的版本情况
最新推荐文章于 2023-05-28 16:29:44 发布