onxxruntime使用cuda以及tensorrt进行加速

 1、版本匹配

版本需要匹配,不然运行会报错

 2、onnxruntime之tensorrt加速

方式一:

OrtTensorRTProviderOptions trt_options{};	
trt_options.trt_max_workspace_size = 2147483648;
trt_options.trt_max_partition_iterations = 10;
trt_options.trt_min_subgraph_size = 5;
trt_options.trt_fp16_enable = 0;
trt_options.trt_int8_enable = 1;
trt_options.trt_int8_use_native_calibration_table = 0;
trt_options.trt_engine_cache_enable = 1;
//trt_options.trt_engine_cache_path = "cache"
trt_options.trt_dump_subgraphs = 1;  
sessionOptions.AppendExecutionProvider_TensorRT(trt_options);

方式二:

OrtSessionOptionsAppendExecutionProvider_CUDA(sessionOptions, 0);
OrtSessionOptionsAppendExecutionProvider_Tensorrt(sessionOptions, 0);

方式三:

const auto& api = Ort::GetApi();
OrtTensorRTProviderOptionsV2* tensorrt_options = nullptr;
api.CreateTensorRTProviderOptions(&tensorrt_options);
std::unique_ptr<OrtTensorRTProviderOptionsV2, decltype(api.ReleaseTensorRTProviderOptions)> rel_trt_options(
    tensorrt_options, api.ReleaseTensorRTProviderOptions);
std::vector<const char*> keys{"device_id", "trt_fp16_enable", "trt_int8_enable", "trt_engine_cache_enable","trt_engine_cache_path"};
std::vector<const char*> values{"0", "1", "0", "1","trt_engine_cache_path"};
api.UpdateTensorRTProviderOptions(rel_trt_options.get(), keys.data(), values.data(), keys.size());
api.SessionOptionsAppendExecutionProvider_TensorRT_V2(static_cast<OrtSessionOptions*>(sessionOptions),
                                                   rel_trt_options.get());

 

 

相关推荐

  1. [12] 使用 CUDA 加速排序算法

    2024-01-06 13:48:05       8 阅读
  2. pytorch导出为onnx,使用onnxruntime进行推理

    2024-01-06 13:48:05       27 阅读
  3. 如何使用cuda进行图像矫正

    2024-01-06 13:48:05       13 阅读

最近更新

  1. TCP协议是安全的吗?

    2024-01-06 13:48:05       18 阅读
  2. 阿里云服务器执行yum,一直下载docker-ce-stable失败

    2024-01-06 13:48:05       19 阅读
  3. 【Python教程】压缩PDF文件大小

    2024-01-06 13:48:05       19 阅读
  4. 通过文章id递归查询所有评论(xml)

    2024-01-06 13:48:05       20 阅读

热门阅读

  1. “人家赚那么多”系列03:如何慢速练习?

    2024-01-06 13:48:05       44 阅读
  2. 切换列表html

    2024-01-06 13:48:05       31 阅读
  3. 【深度学习】神经正切核(NTK)理论

    2024-01-06 13:48:05       30 阅读
  4. 如何理解函数是一等公民?

    2024-01-06 13:48:05       32 阅读
  5. css、js、vue常考部分面试题

    2024-01-06 13:48:05       35 阅读
  6. Leetcode349和350 两个数组的交集问题的思考

    2024-01-06 13:48:05       35 阅读
  7. from pydantic import BaseModel 这个BaseModel是什么意思

    2024-01-06 13:48:05       37 阅读
  8. 苹果MacOS、linux系统如何安装Homebrew

    2024-01-06 13:48:05       45 阅读
  9. opencv期末练习题(7)附带解析

    2024-01-06 13:48:05       38 阅读