Pytorch2TensorRT:pytorch原始码解析生成tensorrt引擎 源码
Atlab SDK团队推理库 运行环境 参考Dockerfile 安装 git clone --recursive https://github.com/huajianni666/Pytorch2TensorRT.git && cd Pytorch2TensorRT wget http://pq16f3soz.bkt.clouddn.com/Pytorch2Trt3rdparty.zip unzip Pytorch2Trt3rdparty.zip && rm Pytorch2Trt3rdparty.zip 编绎 cd python/plugin && mkdir build && cd build && cmake .. make cd ../../ 运行 python refinedet.py
文件列表
Pytorch2TensorRT-master.zip
(预估有个32文件)
Pytorch2TensorRT-master
.gitmodules
170B
example
resnet18_test.cpp
818B
CMakeLists.txt
181B
Dockerfile
3KB
python
common.py
2KB
test.jpg
466KB
refinedet.py
5KB
暂无评论