如何使用 CPU 运行 pytorch 项目？

Question

Pytorch 项目应该在 GPU 上运行。我想在我的笔记本电脑上只用 CPU 运行它。有很多地方在模型、张量等上调用.cuda()，在cuda不可用时执行失败。是否可以在不到处更改代码的情况下做到这一点？

Answer 1

这是我能想到的最简单的解决方法：

1. 将以下行放在代码顶部附近：

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

1. 进行全局替换。将.cuda() 更改为.to(device)，其中device 是步骤1 中设置的变量。

Answer 2

这完全取决于相关项目中代码的结构和设计。通常会有一个参数来指定在哪个设备上运行。你应该检查你的代码。也不建议在您的笔记本电脑 CPU 上运行它，特别是如果项目包含一个中等大小的模型。祝你好运！