deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装

下载对应的tensorrt解压包:
deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装_第1张图片
执行命令:

tar xzvf TensorRT-6.0.1.5.Ubuntu-16.04.x86_64-gnu.cuda-10.1.cudnn7.6.tar.gz

解压后得到对应的文件夹的内容如下:
deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装_第2张图片
激活虚拟环境(仅安装了pytorch):

sourace activate pytorch_1.4

分别进入python、uff、graphsurgeon安装对应的.whl包。后两个需要安装tensorflow,可以不安装对应的uff.whl、graphsurgeon.whl,安装也没事。
deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装_第3张图片
再分别安装一些其他包:

pip install onnx
pip install Pillow
pip install pycuda

deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装_第4张图片
验证,报错:
deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装_第5张图片
添加环境变量:

sudo gedit ~/.bashrc
source ~/.bashrc 

在这里插入图片描述
再次验证:
deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装_第6张图片
完成安装,KOKOKO!!!
c++测试:
进入samples 文件夹下,修改Makefile.config的如下两行,去掉问号(?)
deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装_第7张图片

sudo make 将所有c++的例子编译完成

在这里插入图片描述
deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装_第8张图片
再进入bin文件夹下可以看到编译好的运行文件
deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装_第9张图片
pycharm使用tensorrt

LD_LIBRARY_PATH               /media/chl/workfiles/soft/cuda10.1/TensorRT-6.0.1.5/lib

deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装_第10张图片
deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装_第11张图片

参考链接:

https://www.cnblogs.com/rickyngmo22/p/11650994.html(安装过程)
https://blog.csdn.net/zt1091574181/article/details/100972562
https://blog.csdn.net/calmuse/article/details/93616720(环境变量)
https://www.cnblogs.com/rickyngmo22/p/11650994.html(pycharm 虚拟环境使用)

你可能感兴趣的:(软件安装)