deepin(ubuntu16.04) + tensorrt6.0.1.5 + cuda10.1安装
程序员文章站
2022-04-23 11:20:33
...
下载对应的tensorrt解压包:
执行命令:
tar xzvf TensorRT-6.0.1.5.Ubuntu-16.04.x86_64-gnu.cuda-10.1.cudnn7.6.tar.gz
解压后得到对应的文件夹的内容如下:
**虚拟环境(仅安装了pytorch):
sourace activate pytorch_1.4
分别进入python、uff、graphsurgeon安装对应的.whl包。后两个需要安装tensorflow,可以不安装对应的uff.whl、graphsurgeon.whl,安装也没事。
再分别安装一些其他包:
pip install onnx
pip install Pillow
pip install pycuda
验证,报错:
添加环境变量:
sudo gedit ~/.bashrc
source ~/.bashrc
再次验证:
完成安装,KOKOKO!!!
c++测试:
进入samples 文件夹下,修改Makefile.config的如下两行,去掉问号(?)
sudo make 将所有c++的例子编译完成
再进入bin文件夹下可以看到编译好的运行文件
参考链接:
https://www.cnblogs.com/rickyngmo22/p/11650994.html(安装过程)
https://blog.csdn.net/zt1091574181/article/details/100972562
https://blog.csdn.net/calmuse/article/details/93616720(环境变量)
上一篇: Architecture Rules 2.1.0 发布
下一篇: jrebel插件实现热部署