首页 > TAG信息列表 > lstmFC
libtorch1.5加载lstmFC.pt模型到cuda:1,报错
1、情景 pytorch的模型，torch.jit.trace转换成pt文件 然后通过C++加载调用模型； 2、报错内容： terminate called after throwing an instance of 'std::runtime_error' what(): Input and hidden tensors are not at the same device, found input tensor at cuda:1 and