这个屌丝很懒，什么也没留下！

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

作者：你好赵伟 | 2024-07-19 12:34:39

如何是更改代码使其运行只用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】