ubuntu18.04安装tkDNN的问题

ubuntu18.04安装tkDNN的问题

TensorRT6.0.1安装的部分问题

建议还是下载tar版本然后安装，有问题缺什么改改配置啥的就可以了

第一个问题

安装TensorRT

对于 Python 3.x:

$ sudo pip3 install tensorrt-6.x.x.x-cp3x-none-linux_x86_64.whl

对于这个我开始安装的ubuntu20.04不行，ubuntu20.04默认python3.8,而且没有python2

我用了很多方法，最后改的阮连接，python3显示版本是python3.7了，但还是不行

最后无奈回归ubuntu 18.04(默认python3.6.9)

 

第二个问题就是最后验证是否安装成功tensorrt,uff这是首先进行python

然后出现了问题

1.   [endif]ImportError: libnvinfer.so.6: cannot open shared object file: Nosuch file or directory
2. ImportError: libnvonnxparser.so.6: cannot open shared object file:No such file or directory
3. ImportError: libnvonnxparser_runtime.so.6: cannot open shared objectfile: No such file or directory
4. ImportError: libnvparsers.so.6: cannot open shared object file: Nosuch file or directory
5. ImportError: libnvinfer_plugin.so.6: cannot open shared object file:No such file or directory
6. ImportError: libnvrtc.so.9.0: cannot open shared object file:No such file or directory

这六个问题是解决了一个另一个出来了，六个都解决就没问题了

解决办法：

1.  sudo cp TensorRT-6.01/targets/x86_64-linux-gnu/lib/libnvinfer.so.6  /usr/lib/
2. sudo cp TensorRT-6.01/targets/x86_64-linux-gnu/lib/libnvonnxparser.so.6 /usr/lib/
3.  sudo cp TensorRT-6.01/targets/x86_64-linux-gnu/lib/libnvonnxparser_runtime.so.6  /usr/lib/
4.  sudo cp TensorRT-6.01/targets/x86_64-linux-gnu/lib/libnvparsers.so.6  /usr/lib/
5.  sudo cp TensorRT-6.01/targets/x86_64-linux-gnu/lib/ libnvinfer_plugin.so.6  /usr/lib/
6. sudo cp /usr/local/cuda-9.0/lib64/libnvrtc.so.9.0  /usr/lib/

tkDNN的问题

1.今天查找一个错误，直接复制那个错误查找的没有错误+是什么相关的查找的精确

 

2.tdknn要求下载的darknet是https://git.hipert.unimore.it/fgatti/darknet.git的

同时下载下来之后因为我们要make,根据官方要在makefile中保留GPU = 0

同时在make之前我们就要将weight文件复制过来才行。(步骤不能错)

 

./darknet export <path-to-cfg-file> <path-to-weights> layers这步<..>代表相应的路径

eg:./darknet export /home/haojie/桌面/darknet/cfg/yolov4.cfg /home/haojie/桌面/darknet/yolov4.weights layers(这是我的路径，我将这个darknet放在桌面上)

2.在tkDNN的build文件夹下新建一个你相关项目的文件夹。比如yolo4（这个名称我们具体看）

tkDNN中tests文件夹下找到自己使用的框架。我这次用的是darknet然后进入找到我用的

yolo4.cpp里面有这行代码 std::string bin_path = "yolo4";

所以我们要那样做的。

 

3.我们创建.rt文件

也是在tkDNN的build文件夹下打开命令行窗口。

 

4.在用make编译tkDNN的过程中

git clone https://git.hipert.unimore.it/fgatti/darknet.git

cd darknet

make

mkdir layers debug

./darknet export <path-to-cfg-file> <path-to-weights> layers

 

第三步make的过程中我犯了

install dir:/usr/local
CMake Error: The following variables are used in this project, but they are set to NOTFOUND.
Please set them or make sure they are set and tested correctly in the CMake files:
CUDA_nvinfer_LIBRARY
linked by target "tkDNN" in directory /home/zengxiaojia/ClionProjects/tkDNN

首先我在https://github.com/ceccocats/tkDNN/issues/62查找到了这个问题（好多人犯了同样的错误）

这是系统找不到nvinfer模块; 解决：去到TensorRT安装目录，执行： sudo cp ./lib/libnvinfer.so /usr/lib sudo cp ./include/* /usr/include/ #把所有头文件都拷到/usr/include中去。