欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-04-23 11:20:33
...

下载对应的tensorrt解压包:
deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.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安装
**虚拟环境(仅安装了pytorch):

sourace activate pytorch_1.4

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

pip install onnx
pip install Pillow
pip install pycuda

deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装
验证,报错:
deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装
添加环境变量:

sudo gedit ~/.bashrc
source ~/.bashrc 

deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装
再次验证:
deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装
完成安装,KOKOKO!!!
c++测试:
进入samples 文件夹下,修改Makefile.config的如下两行,去掉问号(?)
deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装

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

deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装
deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装
再进入bin文件夹下可以看到编译好的运行文件
deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装
参考链接:

https://www.cnblogs.com/rickyngmo22/p/11650994.html(安装过程)
https://blog.csdn.net/zt1091574181/article/details/100972562
https://blog.csdn.net/calmuse/article/details/93616720(环境变量)
相关标签: 软件安装