grimoire/amirstan_plugin
Fork: 38 Star: 156 (更新于 2024-11-02 18:30:05)
license: MIT
Language: C++ .
Useful tensorrt plugin. For pytorch and mmdetection model conversion.
最后发布版本: v0.5.0 ( 2021-08-29 22:18:21)
Amirstan_plugin
Amirstan plugin contains some useful TensorRT plugins. These plugins are used to support other projects:
Requirement
- TensorRT >= 8.0
Installation
- Install TensorRT: NVIDIA TensorRT
From sources
-
clone the repo and create
build
foldergit clone --depth=1 https://github.com/grimoire/amirstan_plugin.git cd amirstan_plugin git submodule update --init --progress --depth=1 mkdir build cd build
-
build the project with:
-
plugins only
cmake -DTENSORRT_DIR=${path_to_tensorrt} .. make -j$(nproc)
-
or with DeepStream support
cmake -DTENSORRT_DIR=${path_to_tensorrt} -DWITH_DEEPSTREAM=true -DDeepStream_DIR=${path_to_deepstream} .. make -j$(nproc)
-
-
set the envoirment variable(in ~/.bashrc):
export AMIRSTAN_LIBRARY_PATH=<amirstan_plugin_root>/build/lib
Using Conan
-
Install Conan:
pip install conan
-
Register grimoire's Conan remote:
conan remote add grimoire https://grimoire.jfrog.io/artifactory/api/conan/grimoire-conan
-
Add a
conanfile.txt
file to your project's root with the following content:[requires] amirstan_plugin/0.5.0 [generators] cmake
-
Additionaly, you can add a few options under the [options] section to configure your build:
- tensorrt_dir: path where TensorRT is located. Default
~/SDK/TensorRT
. - with_deepstream: whether to compile with deepstream support. Default
False
. - deepstream_dir: path where deepstream is located. Default
/opt/nvidia/deepstream/deepstream
- cub_root_dir: Default
./third_party/cub
- cuda_arch: list of CUDA architectures to compile for. Default
61;62;70;72;75;80;86
For example, to use a custom TensorRT dir and compile for a specific CUDA architecture:
[requires] amirstan_plugin/0.5.0 [generators] cmake [options] amirstan_plugin:tensorrt_dir=/usr/include/x86_64-linux-gnu amirstan_plugin:cuda_arch=75
- tensorrt_dir: path where TensorRT is located. Default
-
Add the following lines to your project root's
CMakeLists.txt
:INCLUDE(${CMAKE_BINARY_DIR}/conanbuildinfo.cmake) CONAN_BASIC_SETUP()
-
Add conan libs to the linking stage:
target_link_libraries(trt_sample PUBLIC ${CONAN_LIBS} ${CUDA_LIBRARIES} ${CMAKE_THREAD_LIBS_INIT} ${TensorRT_LIBRARIES})
-
Compile your project:
mkdir build cd build conan install .. -s compiler.libcxx=libstdc++11 --build=missing cmake .. make -j$(nproc)
最近版本更新:(数据更新于 2024-09-12 06:30:47)
2021-08-29 22:18:21 v0.5.0
2021-07-24 00:20:47 v0.4.1
2021-04-01 23:46:55 v0.4.0
2020-11-08 19:53:52 v0.3.0
2020-09-26 12:32:38 v0.2.0
2020-09-02 11:03:27 v0.1.0
主题(topics):
tensorrt
grimoire/amirstan_plugin同语言 C++最近更新仓库
2024-11-05 23:57:44 PCSX2/pcsx2
2024-11-05 22:06:04 LizardByte/Sunshine
2024-11-05 00:42:13 ClickHouse/ClickHouse
2024-11-04 21:49:30 notepad-plus-plus/notepad-plus-plus
2024-11-03 22:31:09 MaaAssistantArknights/MaaAssistantArknights
2024-11-02 20:28:28 AaronFeng753/Waifu2x-Extension-GUI