- 博客(4)
- 收藏
- 关注
原创 配置Tusimple的MXNet踩坑记
首先就是从Tusimple的Github开源项目中的INSTALL.md进行,发现里面有Setup with Docker$HOST-SIMPLEDET-DIR:$CONTAINER-WORKDIR registry.cn-beijing.aliyuncs.com/rogerchen/simpledet:cuda9 zsh 首先要安装nvidia-dockerwget -P /...
2019-08-28 15:55:30 368
原创 Mask_RCNN的TensorRT加速
现有的网络上,主流的Mask RCNN主要有两个,一个是matterport的Keras版本https://github.com/matterport/Mask_RCNN,有13.5K的star,一个是facebookresearch的PyTorch版本6.2K的star,链接如下https://github.com/facebookresearch/maskrcnn-bench...
2019-08-14 13:50:59 6871 24
原创 Pytorch转Onnx转TensorRT踩坑记
转换Onnx过程中:PyTorch v1.0.1 Reshape不支持报错源码安装PyTorch v1.0.1 caffe2报错PyTorch v1.2.0 需要升级cuda10.0以上PyTorch v1.1.0报错:RuntimeError: Failed to export an ONNX attribute, since it's not constant, ple...
2019-08-13 17:25:20 7812
原创 Onnx-TensorRT安装指南(踩坑记)
首先要来到onnx-tensorrt的Github官网Github链接 下面就是步骤git clone --recursive https://github.com/onnx/onnx-tensorrt.gitmkdir buildcd buildcmake .. -DTENSORRT_ROOT=<tensorrt_install_dir&g...
2019-08-07 16:28:25 12143 2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人