RuntimeError: Input, output and indices must be on the current device

本文主要是介绍RuntimeError: Input, output and indices must be on the current device，对大家解决编程问题具有一定的参考价值，需要的程序猿们随着小编来一起学习吧！

Traceback (most recent call last):
  File "detect.py", line 236, in <module>
    detect()
  File "detect.py", line 47, in detect
    torch.onnx.export(model,               # model being run
  File "/home/zjkj/miniconda3/envs/yolov5/lib/python3.8/site-packages/torch/onnx/__init__.py", line 271, in export
    return utils.export(model, args, f, export_params, verbose, training,
  File "/home/zjkj/miniconda3/envs/yolov5/lib/python3.8/site-packages/torch/onnx/utils.py", line 88, in export
    _export(model, args, f, export_params, verbose, training, input_names, output_names,
  File "/home/zjkj/miniconda3/envs/yolov5/lib/python3.8/site-packages/torch/onnx/utils.py", line 694, in _export
    _model_to_graph(model, args, verbose, input_names,
  File "/home/zjkj/miniconda3/envs/yolov5/lib/python3.8/site-packages/torch/onnx/utils.py", line 502, in _model_to_graph
    params_dict = torch._C._jit_pass_onnx_constant_fold(graph, params_dict,
RuntimeError: Input, output and indices must be on the current device

这篇关于RuntimeError: Input, output and indices must be on the current device的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！