- CUDA: 11.6
- TENSORRT: TensorRT-8.6.1.6
mkdir build
cd build
cmake ..
make -j8
make installcd test
# 生成trt engine并对比trt算子结果与torch算子结果
python3 torch_matmul_to_trt.py
# 生成trt engine并对比trt算子结果与tf算子结果
python3 tf_matmul_to_trt.pytrt_custom_op
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|
parent directory.. | ||||