我們可以在帶有GPU的機器上訓練帶有手電筒的模型，並僅在具有CPU的機器上使用此模型？

Question

我正在使用與cuda一起的火炬後端在帶有GPU的機器上訓練模型。在建立模型後，我可以使用這種模式對帶CPU的機器進行推理嗎？

Answer 1

是的，你可以保存你的模型之前，記得運行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())