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

2022-09-30 10:05:49

查看Pytorch版本

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

查看Pytorch是否支持GPU加速

import torchprint(torch.cuda.is_available())>>> True #这便代表此Pytorch支持GPU加速
  • 作者：SoaringPigeon
  • 原文链接：https://blog.csdn.net/weixin_45429089/article/details/116170370
    更新时间：2022-09-30 10:05:49