欢迎光临散文网 会员登陆 & 注册

TensorRt部署 cublas/cublasLtWrapper.cpp:27

2023-11-09 11:31 作者:你晗真好看  | 我要投稿

cublas/cublasLtWrapper.cpp:279 确认环境无误后,将转模型命令变为  ./trtexec --onnx=/home/py/code/JDAI/fast-reid/tools/deploy/onnx_model/baseline.onnx --tacticSources=

-cublasLt,+cublas

--workspace=2048 --fp16 --saveEngine=/home/py/code/JDAI/fast-reid/tools/deploy/onnx_model/baseline2.trt

TensorRt部署 cublas/cublasLtWrapper.cpp:27的评论 (共 条)

分享到微博请遵守国家法律