ross / trading journal
← 全部文章

onnxruntime InferenceSession Segmentation fault 解决方案

技术分享
type
Post
status
Published
date
Apr 22, 2022
slug
ONNX Segmentation fault solver
summary
onnxruntime1.7-1.9 在 InferenceSession 初始化时候有些记性会出现 segmentation fault,这个可能和多线程执行有关。
tags
debug log
category
技术分享
icon
password
URL
onnxruntime1.7-1.9 在 InferenceSession 初始化时候有些记性会出现 segmentation fault,这个可能和多线程执行有关。目前1.10版本没有出现这个问题。
网上也有一些方法可以尝试:
 
import onnxruntime as ort sess_options = ort.SessionOptions() sess_options.intra_op_num_threads = 8 sess = ort.InferenceSession('some_model.onnx', sess_options=sess_options)
选择合适的intra_op_num_threads即可。

评论

Behind the trades.

我是 Ross,交易员,也是开发者。
研究市场微观结构、量化策略与 AI。记录假设、执行与复盘,让每一次决策都有迹可循。