栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

ubuntu瀹夎tensorflow cpu(ubuntu瀹夎tensorflow1.14)

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

ubuntu瀹夎tensorflow cpu(ubuntu瀹夎tensorflow1.14)

ubuntu安装tensorrt
前提:先安装好cuda和cudnn
步骤一:下载tensorrt压缩包。找到自己对应cuda版本的安装包
从nvidia的官网下载
https://developer.nvidia.com/nvidia-tensorrt-download
(由于我安装的是cuda11,所以我这里下载的是下面这个)

步骤二:下载下来的tensorrt压缩包为TensorRT-8.0.1.6.Linux.x86_64-gnu.cuda-11.3.cudnn8.2.tar.gz,这里放置在服务器的一个路径下,(我是放在了/opt下面)。然后解压这个文件。

cd /opt
tar -zxvf TensorRT-8.0.1.6.Linux.x86_64-gnu.cuda-11.3.cudnn8.2.tar.gz

步骤三:添加环境变量

sudo vi ~/.bashrc
# 在最后添加一行
export LD_LIBRARY_PATH=/opt/TensorRT-8_0_1_6/lib:$LD_LIBRARY_PATH
source ~/.bashrc

步骤四:安装tensorrt的python包。进入tensorrt压缩之后的文件夹,然后进入里面的python文件夹

cd /opt/TensorRT-8_0_1_6/python

python文件夹里面有很多版本,使用pip安装自己对应的python版本

pip install tensorrt-8.0.1.6-cp35-none-linux_x86_64.whl

顺利的话,这时候tensorrt包应该已经安装好了
进入python,导入一下tensorrt包试试

这就安装成功了!!!

可以顺便安装其他两个包
1)安装 uff 包

cd ../uff    # 切换到 uff 文件夹
pip install uff-0.5.5-py2.py3-none-any.whl

测试是否安装成功:

which convert-to-uff  # 会输出安装路径

2)安装 graphsurgen 包

cd ../graphsurgeon     # 切换到 graphsurgeon 文件夹
pip install graphsurgeon-0.3.2-py2.py3-none-any.whl

参考链接:
https://developer.nvidia.com/nvidia-tensorrt-8x-download
https://blog.csdn.net/xulei_Tao/article/details/90339694

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/772603.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号