search

NVIDIA / trt-samples-for-hackathon-cn

Simple samples for TensorRT programming
Apache License 2.0
1.47k stars 337 forks source link

07-Tool/trtexec/command.sh fail #99

Closed bolin12 closed 4 months ago

bolin12 commented 9 months ago

07-Tool/trtexec/result-03.log：

&&&& RUNNING TensorRT.trtexec [TensorRT v8601] # trtexec model-02.plan --trt --shapes=tensorX:4x1x28x28 --noDataTransfers --useSpinWait --useCudaGraph --verbose [11/27/2023-08:17:03] [E] Model missing or format not recognized

wanderHZ commented 9 months ago

You can change the code in the command.sh to the following format:

# 03-Load TensorRT engine built above and do inference
trtexec \
    --loadEngine=model-02.plan \
    --shapes=tensorX:4x1x28x28 \
    --noDataTransfers \
    --useSpinWait \
    --useCudaGraph \
    --verbose \
    > result-03.log 2>&1