- cuda-11
- cudnn-8.2.1
- tensorrt 8.2.1.8
git clone https://github.com/NVIDIA/TensorRT/
pushd TensorRT
git submodule update --init --recursive
git checkout 156c59ae86d454fa89146fe65fa7332dbc8c3c2b
git submodule update
git apply ../tensorrt.diff
popd
pushd TensorRT
mkdir -p build && cd build
CUDACXX=/opt/anaconda3/envs/cuda-11/bin/nvcc cmake .. -DTRT_LIB_DIR=/opt/anaconda3/envs/cuda-11/lib -DTRT_OUT_DIR=`pwd`/out
make
popd
get models from https://mega.nz/folder/X5UD1LDY#8ZI-gAq6AkpcUbE2z4n6RA and save
models to model
directory
make run-mnist make run-googlenet make run-mobilenet make run-resnet
- turn on
CPPFLAGS += -DINT8
in Makefile - make clean
- make run-mnist