import torch
print(torch.__version__)
print(torch.cuda.is_available())
1. 查看torch版本
>>> import torch
>>> print(torch.__version__)
1.2.0
2.查看cuda版本
>>> print(torch.version.cuda)
10.0
3.查看cudnn版本
>>> print(torch.backends.cudnn.version())
7401
————————————————
版权声明:本文为CSDN博主「请让我按时毕业」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/weixin_45819759/article/details/130477297