Deep learning 我们可以在带有GPU的机器上训练一个带有torch的模型，并在只有CPU的机器上使用这个模型吗？

我正在使用torch后端和cuda在带有GPU的机器上训练模型。建立模型后，我可以使用此模式在仅使用CPU的机器上进行推理吗？

是的，在保存模型之前，请记住运行

model:double（）

将模型权重从Cuda张量转换为double张量

require 'cutorch';
require 'cunn';
require 'nn';
model = nn.Sequential()
model:add(nn.Linear(10, 20))


print(model:parameters())
model:cuda();
print(model:parameters())
model:double();
print(model:parameters())

---

是的，您可以在保存模型之前，记得运行

model:double（）

将模型权重从Cuda张量转换为双张量

require 'cutorch';
require 'cunn';
require 'nn';
model = nn.Sequential()
model:add(nn.Linear(10, 20))


print(model:parameters())
model:cuda();
print(model:parameters())
model:double();
print(model:parameters())