(修改为) .to(device)..._如何是更改代码使其运行只用cpu">
当前位置:   article > 正文

将代码中的Pytorch写的gpu运行代码改写成为cpu版本_如何是更改代码使其运行只用cpu

如何是更改代码使其运行只用cpu

1、修改torch.load,添加选择cpu

torch.load(opt.model, map_location='cpu')
  • 1

2、将cuda()修改,首先用device判断一下torch是否可用,不行就用cpu,将原来直接写的.cuda()写成如下:

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
.cuda()  -->(修改为) .to(device)
  • 1
  • 2
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/你好赵伟/article/detail/851514?site
推荐阅读
相关标签
  

闽ICP备14008679号