查看Pytorch版本以及是否支持GPU加速

Posted 2022-12-12 SoaringPigeon

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了查看Pytorch版本以及是否支持GPU加速相关的知识，希望对你有一定的参考价值。

查看Pytorch版本

import torch
print(torch.__version__)
>>> 1.4.0 #这里便是torch版本

import torch
print(torch.cuda.is_available())
>>> True #这便代表此Pytorch支持GPU加速

以上是关于查看Pytorch版本以及是否支持GPU加速的主要内容，如果未能解决你的问题，请参考以下文章