当前位置:   article > 正文

快速将需要GPU计算的代码转为CPU计算_torch算子移到cpu

torch算子移到cpu
  •  在进行深度学习训练时,下载的代码有的需要GPU计算,而电脑又没有GPU,此时可以在代码开头添加如下代码获取当前计算机是否有GPU

# gpu or not
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

  • 然后在代码中 用 .to(device)替换.cuda() 或者.cpu() 即可在有GPU的计算机上通过GPU计算加速,在没有GPU的计算机上用CPU计算,不过速度较慢

  • 在写深度学习算法时,也尽可能采用这种同样的代码,方便在任何机器上运行。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小惠珠哦/article/detail/851428
推荐阅读
相关标签
  

闽ICP备14008679号