![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
阔岩
这个作者很懒,什么都没留下…
展开
-
TensorRT使用基本流程
tensorRT的使用包括两个阶段, build and runtime (deployment):build:Import and optimize trained models to generate inference enginesbuild阶段主要完成模型转换(从caffe或TensorFlow到TensorRT),在模型转换时会完成前述优化过程中的层间融合,精度校准。这一步的输出...原创 2020-01-14 10:15:01 · 4909 阅读 · 4 评论 -
TensorRT安装和Hello World
安装1. 官网下载tensorRT的安装包,解压即可,使用时在CMakeLists里指定头文件和库文件位置即可。不推荐使用deb文件安装,因为它要求配置文件也是deb文件安装。2. 下载对应的cuda版本和cudnn版本,cuda版本一定要严格对应。我的配置是ubuntu16.04,cuda9.0,cudnn7.3cudnn版本我没使用官网指定的7.6,虽然会警告,程序还是可以运行。He...原创 2020-01-14 10:08:14 · 1529 阅读 · 0 评论 -
caffe python 接口,ImportError, No module named caffe
这种情况一般是没有把caffe中的和python相关的内容的路径添加到python的编译路径中,或者caffe的pycaffe没有编译1.首先编译pycaffecaffe build文件夹下编译pycaffecd caffe/build/make pycaffe2.然后添加caffe的python搜索路径2.1官方解决方案:临时解决:终端中输入export PYTHONPAT...原创 2020-01-10 19:33:17 · 155 阅读 · 0 评论