torch::jit::load 参数

Libtroch模型在GPU上但是推理在CPU进行，GPU不动

• 建议你看下这篇博客👉 ：“load”: 不是“torch::jit”的成员

看下你的gpu的batch size是不是太大。