【tf-trt环境配置】tensorflow+cuda+cudnn+tensorrt安装记录

已于 2022-02-15 16:21:37 修改
阅读量4.7k 收藏 8
点赞数 1
分类专栏： 软件，库，环境配置安装 文章标签： tensorflow 深度学习 python
于 2021-01-27 21:35:19 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_44876051/article/details/113267175
版权

本文将介绍如何编译tensorflow+cuda+cudnn+tensorrt

环境查看

tensorflow版本要求：https://www.tensorflow.org/install/source
在这里插入图片描述
TennsorRT环境要求：
https://docs.nvidia.com/deeplearning/tensorrt/archives/index.html
选择
在这里插入图片描述

在这里插入图片描述
综合对比确定环境如下：

项目Value
cuda10.1 update2
cudnn7.6.5
bazel3.1.0
tensorrt6.0.1
tensorflow2.3.0
gcc g++7.3
numpy<1.19

一：安装cuda10.1 update2

注意：需要安装最新NVIDIA驱动
下载地址：https://developer.nvidia.com/cuda-toolkit-archive
在这里插入图片描述
执行代码：

wget https://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
sudo sh cuda_10.1.243_418.87.00_linux.run

配置CUDA相关环境变量
修改/.bashrc文件（或/.zhsrc） 在文件末尾添加环境变量

Tensorflow官方安装历程要求注意的是:配置PATH和LD_LIBRARY_PATH和CUDA_HOME环境变量.
 
vim ~/.bashrc #修改配置文件（如果你用的是zsh，则需要修改 ~/.zshrc文件）
 
#在文件结尾处添加
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=/usr/local/cuda

补充：

  1. 若文件下载较慢，可通过以下命令设置终端代理
export http_proxy=http://127.0.0.1:12333
export https_proxy=http://127.0.0.1:12333
  1. 若你的系统中存在多个cuda版本，可通过以下教程进行cuda版本切换https://blog.csdn.net/yinxingtianxia/article/details/80462892 https://blog.csdn.net/wuprogrammer/article/details/108942268

二：cudnn安装

下载地址：https://developer.nvidia.com/rdp/cudnn-archive
在这里插入图片描述
然后解压，并进入到相应目录，运行以下命令：

sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

在这里插入图片描述
PS:cudnn版本查询：

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

PS2：cudnn8 中的部分安装步骤有所改动，请读者参考最新的安装教程https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html

三：GCC版本切换

执行以下命令查看gcc版本

gcc --version

若gcc版本不符合要求：
按照以下顺序下载相应版本：

  1. 使用以下命令将 ubuntu-toolchain-r/test PPA添加到 Ubuntu 系统：
sudo apt install software-properties-common
sudo add-apt-repository ppa:ubuntu-toolchain-r/test
  1. 执行以下命令安装所需的 GCC 和 G ++ 版本：
    (如果无法定位安装包，可以在设置里换个安装源）
sudo apt install gcc-7 g++-7

以下命令将为每个版本配置替代方案并将优先级与其关联，默认版本是具有最高优先级的版本，在我们的例子中是 gcc-7：

sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 90 --slave /usr/bin/g++ g++ /usr/bin/g++-7
  1. 如果要更改默认版本，可以使用 update-alternatives 命令：
sudo update-alternatives --config gcc

四：安装tensorRT

下载地址：
https://developer.nvidia.com/zh-cn/tensorrt

解压文件夹

在.bashrc文件里添加绝对路径

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:<eg:TensorRT-6.x.x.x/lib>

终端进入相应文件夹（若由anaconda虚拟环境，则在虚拟环境下执行以下安装命令）：
1.安装TensorRT wheel文件

cd TensorRT-6.x.x.x/python

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

2.安装uff

cd TensorRT-6.x.x.x/uff

sudo pip3 install uff-0.6.5-py2.py3-none-any.whl

3.安装graphsurgeon ：

cd TensorRT-6.x.x.x/graphsurgeon

sudo pip3 install graphsurgeon-0.4.1-py2.py3-none-any.whl

五：安装bazel

下载地址：https://github.com/bazelbuild/bazel/releases

chmod +x bazel-<version>-installer-linux-x86_64.sh
./bazel-<version>-installer-linux-x86_64.sh --user

在.bashrc文件夹中添加环境变量

export PATH="$PATH:$HOME/bin"

六：安装tensorflow

若有虚拟环境，进入虚拟环境后执行以下命令
安装 预处理包

pip install -U --user pip  wheel
pip install numpy==1.16
pip install -U --user keras_preprocessing --no-deps

拉取tensorflow源码

git clone https://github.com/tensorflow/tensorflow.git

cd tensorflow

git checkout r2.3

./configure

接下来，配置系统会给出各种询问，以确认编译时的配置参数：
参数具体解释：https://blog.csdn.net/yhily2008/article/details/79967118

spicker@spicker-1:~/software/tensorflow$ ./configure 
WARNING: --batch mode is deprecated. Please instead explicitly shut down your Bazel server using the command "bazel shutdown".
You have bazel 0.25.2 installed.
Please specify the location of python. [Default is /home/spicker/anaconda3/bin/python]: 


Found possible Python library paths:
  /home/spicker/anaconda3/lib/python3.6/site-packages
Please input the desired Python library path to use.  Default is [/home/spicker/anaconda3/lib/python3.6/site-packages]

Do you wish to build TensorFlow with XLA JIT support? [Y/n]: n
No XLA JIT support will be enabled for TensorFlow.

Do you wish to build TensorFlow with OpenCL SYCL support? [y/N]: N
No OpenCL SYCL support will be enabled for TensorFlow.

Do you wish to build TensorFlow with ROCm support? [y/N]: N
No ROCm support will be enabled for TensorFlow.

Do you wish to build TensorFlow with CUDA support? [y/N]: y
CUDA support will be enabled for TensorFlow.

Do you wish to build TensorFlow with TensorRT support? [y/N]: y
TensorRT support will be enabled for TensorFlow.

Found CUDA 10.1 in:
    /usr/local/cuda/lib64
    /usr/local/cuda/include
Found cuDNN 7.6.3 in:
    /usr/local/cuda/lib64
    /usr/local/cuda/include
Found TensorRT 5 in:
    /usr/lib
    /usr/include


Please specify a list of comma-separated CUDA compute capabilities you want to build with.
You can find the compute capability of your device at: https://developer.nvidia.com/cuda-gpus.
Please note that each additional compute capability significantly increases your build time and binary size, and that TensorFlow only supports compute capabilities >= 3.5 [Default is: 6.1]: 


Do you want to use clang as CUDA compiler? [y/N]: N
nvcc will be used as CUDA compiler.

Please specify which gcc should be used by nvcc as the host compiler. [Default is /usr/bin/gcc]: 


Do you wish to build TensorFlow with MPI support? [y/N]: N
No MPI support will be enabled for TensorFlow.

Please specify optimization flags to use during compilation when bazel option "--config=opt" is specified [Default is -march=native -Wno-sign-compare]: 


Would you like to interactively configure ./WORKSPACE for Android builds? [y/N]: N
Not configuring the WORKSPACE for Android builds.

Preconfigured Bazel build configs. You can use any of the below by adding "--config=<>" to your build command. See .bazelrc for more details.
	--config=mkl         	# Build with MKL support.
	--config=monolithic  	# Config for mostly static monolithic build.
	--config=gdr         	# Build with GDR support.
	--config=verbs       	# Build with libverbs support.
	--config=ngraph      	# Build with Intel nGraph support.
	--config=numa        	# Build with NUMA support.
	--config=dynamic_kernels	# (Experimental) Build kernels into separate shared objects.
Preconfigured Bazel build configs to DISABLE default on features:
	--config=noaws       	# Disable AWS S3 filesystem support.
	--config=nogcp       	# Disable GCP support.
	--config=nohdfs      	# Disable HDFS support.
	--config=noignite    	# Disable Apache Ignite support.
	--config=nokafka     	# Disable Apache Kafka support.
	--config=nonccl      	# Disable NVIDIA NCCL support.
Configuration finished

补充：若config文件找不到cuda，cudnn，tensorRT文件如下图：

felaim@felaim-pc:~/Documents/software/tensorflow-r1.14$ ./configure 
WARNING: --batch mode is deprecated. Please instead explicitly shut down your Bazel server using the command "bazel shutdown".
You have bazel 0.24.1 installed.
Please specify the location of python. [Default is /usr/bin/python]: /home/felaim/anaconda3/envs/tensorflow-gpu/bin/python


Found possible Python library paths:
  /home/felaim/anaconda3/envs/tensorflow-gpu/lib/python3.7/site-packages
Please input the desired Python library path to use.  Default is [/home/felaim/anaconda3/envs/tensorflow-gpu/lib/python3.7/site-packages]

Do you wish to build TensorFlow with XLA JIT support? [Y/n]: n
No XLA JIT support will be enabled for TensorFlow.

Do you wish to build TensorFlow with OpenCL SYCL support? [y/N]: n
No OpenCL SYCL support will be enabled for TensorFlow.

Do you wish to build TensorFlow with ROCm support? [y/N]: n
No ROCm support will be enabled for TensorFlow.

Do you wish to build TensorFlow with CUDA support? [y/N]: y
CUDA support will be enabled for TensorFlow.

Do you wish to build TensorFlow with TensorRT support? [y/N]: y
TensorRT support will be enabled for TensorFlow.

Could not find any NvInferVersion.h matching version '' in any subdirectory:
        ''
        'include'
        'include/cuda'
        'include/*-linux-gnu'
        'extras/CUPTI/include'
        'include/cuda/CUPTI'
of:
        '/home/felaim/intel/ipp/lib/intel64_lin'
        '/lib/x86_64-linux-gnu'
        '/usr'
        '/usr/lib'
        '/usr/lib/x86_64-linux-gnu'
        '/usr/lib/x86_64-linux-gnu/libfakeroot'
        '/usr/lib/x86_64-linux-gnu/mesa'
        '/usr/lib/x86_64-linux-gnu/mesa-egl'
        '/usr/local/cuda'
        '/usr/local/cuda-10.1/targets/x86_64-linux/lib'
        '/usr/local/cuda/extras/CUPTI/lib64'
        '/usr/local/lib'
Asking for detailed CUDA configuration...

Please specify the CUDA SDK version you want to use. [Leave empty to default to CUDA 10]: 


Please specify the cuDNN version you want to use. [Leave empty to default to cuDNN 7]: 


Please specify the TensorRT version you want to use. [Leave empty to  default to TensorRT 5]: 6


Please specify the locally installed NCCL version you want to use. [Leave empty to use http://github.com/nvidia/nccl]: 


Please specify the comma-separated list of base paths to look for CUDA libraries and headers. [Leave empty to use the default]: /usr/local/cuda,/usr/local/cuda/bin,/usr/local/cuda/lib64,/usr/local/cuda/include,/home/felaim/Documents/software/TensorRT-6.0.1.5,/usr/lib/x86_64-linux-gnu/


Found CUDA 10.1 in:
    /usr/local/cuda/lib64
    /usr/local/cuda/include
Found cuDNN 7 in:
    /usr/local/cuda/lib64
    /usr/local/cuda/include
Found TensorRT 6 in:
    /home/felaim/Documents/software/TensorRT-6.0.1.5/lib
    /home/felaim/Documents/software/TensorRT-6.0.1.5/include

Please specify a list of comma-separated CUDA compute capabilities you want to build with.
You can find the compute capability of your device at: https://developer.nvidia.com/cuda-gpus.
Please note that each additional compute capability significantly increases your build time and binary size, and that TensorFlow only supports compute capabilities >= 3.5 [Default is: 5.2]: 


Do you want to use clang as CUDA compiler? [y/N]:   
nvcc will be used as CUDA compiler.

Please specify which gcc should be used by nvcc as the host compiler. [Default is /usr/bin/gcc]: 


Do you wish to build TensorFlow with MPI support? [y/N]: 
No MPI support will be enabled for TensorFlow.

Please specify optimization flags to use during compilation when bazel option "--config=opt" is specified [Default is -march=native -Wno-sign-compare]: 


Would you like to interactively configure ./WORKSPACE for Android builds? [y/N]: 
Not configuring the WORKSPACE for Android builds.

Preconfigured Bazel build configs. You can use any of the below by adding "--config=<>" to your build command. See .bazelrc for more details.
	--config=mkl         	# Build with MKL support.
	--config=monolithic  	# Config for mostly static monolithic build.
	--config=gdr         	# Build with GDR support.
	--config=verbs       	# Build with libverbs support.
	--config=ngraph      	# Build with Intel nGraph support.
	--config=numa        	# Build with NUMA support.
	--config=dynamic_kernels	# (Experimental) Build kernels into separate shared objects.
Preconfigured Bazel build configs to DISABLE default on features:
	--config=noaws       	# Disable AWS S3 filesystem support.
	--config=nogcp       	# Disable GCP support.
	--config=nohdfs      	# Disable HDFS support.
	--config=noignite    	# Disable Apache Ignite support.
	--config=nokafka     	# Disable Apache Kafka support.
	--config=nonccl      	# Disable NVIDIA NCCL support.
Configuration finished

可在配置过程输入文件位置解决，如：

/usr/local/cuda,/usr/local/cuda/bin,/usr/local/cuda/lib64,/usr/local/cuda/include,/usr/lib/x86_64-linux-gnu/,/home/hove/TensorRT-6.0.1.5

编译源码：

若配置较好

bazel build --config=cuda --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" --verbose_failures //tensorflow/tools/pip_package:build_pip_package

较差

bazel build --config=cuda --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0"  --local_ram_resources=4096 --local_cpu_resources=3 --verbose_failures //tensorflow/tools/pip_package:build_pip_package
  1. –cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" 是让gcc5.0以后的版本对源码兼容
  2. –local_ram_resources=4096 --local_cpu_resources=3分别限制编译过程占用的内存和核数
  3. –verbose_failures 将编译过程产生的错误显示出来

在tensorflow文件夹中运行

./bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg

备注:若每次编译失败需要清除bazel缓存，重新编译

rm -rf ~/.cache/bazel/*

安装pip文件：

pip install /tmp/tensorflow_pkg/tensorflow-version-tags.whl

大功告成！

HoveXb
关注
  • 1
    点赞
  • 踩
  • 8
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
专栏目录
nv-tensorrt-repo-ubuntu1604-cuda10.0-trt7.0.0.11-ga-20191216_1-1_amd64.deb
04-13
nv-tensorrt-repo-ubuntu1604-cuda10.0-trt7.0.0.11-ga-20191216_1-1_amd64.deb
TensorRT-Plugin实现之ROIAlignPlugin算子实现-cuda-TRT8.zip
最新发布
05-22
本篇将详细介绍如何在CUDA环境下，针对TensorRT 8实现ROIAlignPlugin算子，这是一个在目标检测任务中广泛使用的算子。 ROIAlign（Region of Interest Align）是 Faster R-CNN 和 Mask R-CNN 等算法中的关键步骤，...
【笔记】解决Ubuntu 23.04下Warning: Could not find TensorRT的问题
miles2007的博客
08-29 1万+
解决问题笔记：TF-TRT Warning: Could not find TensorRT & Cannot dlopen some GPU libraries.
TensorRT7 + Onnx_TensorRT 安装过程记录
史蒂夫方
12-25 1万+
TensorRT + Onnx_TensorRT 安装步骤 我个人是将TRT和Onnx_TensorRT安装在docker的container中， 因为云都是提供的docker环境， 实际使用起来跟一般没有区别 安装TensorRT 依照CUDA版本下载相对应的CUDNN 下载对应cuda版本的cudnn tar -xvzf cudnn-10.0-linux-x64-v7.6.5.32.t...
Docker【部署 07】镜像内安装tensorflow-gpu及调用GPU多个问题处理Could not find cuda drivers+unable to find libcuda
2401_84545016的博客
05-16 504
Python编程学习，学习内容包含：语法、正则、文件、 网络、多线程等常用库，推荐《Python核心编程》，不要看完；在实际的渗透测试过程中，面对复杂多变的网络环境，当常用工具不能满足实际需求的时候，往往需要对现有工具进行扩展，或者编写符合我们要求的工具、自动化脚本，这个时候就需要具备一定的编程能力。恭喜你，如果学到这里，你基本可以从事一份网络安全相关的工作，比如渗透测试、Web 渗透、安全服务、安全分析等岗位；检测CUDA版本是必要的，由于使用的是导出的镜像文件，2和3的方法无法使用，最终使用。
Could not find TensorRT(missing: CUDA_CUBLAS_LIBRARIES)
P0ny的博客
12-12 4485
编译TensorRT相关项目， 报如题的错误 解决办法升级cmake到3.14.5， wget https://cmake.org/files/v3.14/cmake-3.14.5.tar.gz tar -zxvf cmake-3.14.5.tar.gz cd cmake-3.14.5 ./configure make sudo make install cmake -version ...
colab中搭建tf-trt环境——解决TensorRT版本匹配问题
xixi11a的博客
04-13 1824
colab中搭建tf-trt环境——解决TensorRT版本匹配问题 可行的环境版本匹配： Ubuntu18.04 cuda11.1 cudnn8 python3.7 tensorflow-gpu2.7.0 TensorRT7.2.2 pycuda2021.1 查看版本命令： !lsb_release -a " No LSB modules are available. Distributor ID: Ubuntu Description: Ubuntu 18.04.5 LTS Release: 18.
TensorRT cmake时出现Could NOT find TENSORRt library
fanzy1234的博客
08-13 6642
参考大佬们的解决方法 https://blog.csdn.net/zzyczzyc/article/details/93492559 https://blog.csdn.net/haiyinshushe/article/details/84256137 还不行的话（我的安装就是） 我用的远程服务器，不知怎么还是不能解决无法定位软件包的问题 解决方法 // An highlighted block ...
TensorRT-7.2.3.4.Windows10.x86-64.cuda-10.2.cudnn8.1.zip
02-08
8. **与ONNX和TF-TRT的兼容性**：支持导入ONNX（Open Neural Network Exchange）模型，以及与TensorFlow的集成，通过TF-TRT（TensorFlow-TensorRT）将TensorFlow模型转换为TensorRT可优化的格式。 综上所述，...
TensorRT-Yolov5的tensorrt部署-TF2+TFlite+ONNX+TRT-优质算法部署项目实战.zip
02-28
本项目实战旨在深入探讨TensorRT在Yolov5模型上的应用，通过TensorFlow 2 (TF2)、TensorFlow Lite (TFlite)、ONNX（Open Neural Network Exchange）以及NVIDIA的TensorRT (TRT)进行高效部署。这些工具和技术的结合...
TensorRT-Plugin实现之VoxelGenerator算子实现-cuda-TRT8.zip
05-22
在本文中，我们将深入探讨如何使用TensorRT的Plugin机制来实现自定义的VoxelGenerator算子，并结合CUDA进行优化，以适应TensorRT 8版本。 首先，理解TensorRT Plugin是关键。TensorRT Plugin是一种扩展机制，允许...
Docker【部署 07】镜像内安装tensorflow-gpu及调用GPU多个问题处理Could not find cuda drivers+unable to find libcuda.so...
Java与大数据相关实践内容分享！
10-24 1954
Docker镜像内安装tensorflow-gpu及调用GPU多个问题处理Could not find cuda drivers+unable to find libcuda.so...
小白笔记[1]| 运行tf问题整理(持续完善)
热门推荐
xr627的博客
11-01 2万+
整理一些使用tensorflow过程中的警告信息处理
关于tensorRT的失败使用总结
Henry_Avery的博客
08-25 1860
TensorRT真难用啊
Airflow【部署 01】Airflow官网Quick Start实操（一篇学会部署Airflow）_apache airflow教程(1)
2401_84253850的博客
05-07 910
①了解行业相关背景，前景，确定发展方向。②学习网络安全相关法律法规。③网络安全运营的概念。④等保简介、等保规定、流程和规范。（非常重要）①渗透测试的流程、分类、标准②信息收集技术：主动/被动信息搜集、Nmap工具、Google Hacking③漏洞扫描、漏洞利用、原理，利用方法、工具（MSF）、绕过IDS和反病毒侦察④主机攻防演练：MS17-010、MS08-067、MS10-046、MS12-20等①Windows系统常见功能和命令。
TensorRT中的常见问题
专注于人工智能领域的小何尚
02-16 1415
以下部分有助于回答有关 NVIDIA TensorRT 典型用例的最常见问题。
TFX完整流水线示例以及补充（上）
T_eddy的博客
06-27 727
TFX完整流水线示例以及补充（上），包含从ExampleGen到Transform组件。
【解决方案】TensorRT安装方法，及测试TensorRT中sample给出的mnist时，出现Could not find x.pgm错误解决方法
lkh798842577的博客
08-18 1205
环境： ubuntu18.04 pycuda pil tensorflow-gpu=1.14.0 cuda=10.0 cudnn=7.6.5 python=3.6 1.安装 按照大佬给出的方法安装：https://blog.csdn.net/u011681952/article/details/105973996 需要注意的是，tensorRT的deb安装方式要求CUDA也是deb格式安装的，否则容易出错（很多安装CUDA采用的是runfile格式）。 因我采用的安装CUDA方式时runfile方式，故我选
Could not load dynamic library ‘libnvinfer.so.6‘； 错误解决
qq_57076285的博客
12-06 1886
错误： import tensorflow as tf Could not load dynamic library 'libnvinfer.so.6'; dlerror: libnvinfer.so.6: cannot open shared object file: No such file or directory; Cannot dlopen some TensorRT libraries. If you would like to use Nvidia GPU with TensorRT,
TF-TRT Warning: Could not find TensorRT
09-05
你好！关于你提到的 TF-TRT 警告信息，这通常表示 TensorFlow 并未找到安装在系统上的 TensorRT 库。TF-TRT 是一个 TensorFlow 的转换工具，它使用 TensorRT 进行高性能推理。要解决此问题，需要确保已正确安装 TensorRT 并与 TensorFlow 进行了兼容。你可以按照 TensorRT 官方文档提供的步骤进行安装和配置，以便在 TensorFlow 中使用 TF-TRT。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值