TensorRt部署 cublas/cublasLtWrapper.cpp:27
cublas/cublasLtWrapper.cpp:279 確認(rèn)環(huán)境無誤后,將轉(zhuǎn)模型命令變?yōu)??./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
標(biāo)簽: