centos7 安装TensorRT

1.安装cuda、cudnn

问题:

patchlevel.h没有那个文件
将python-devel 改为python3-devel
2.安装tensorRT
tensorRT安装官网
tensorrt
tensorrt 下载
tensorRT sample
TensorRT API 使用

3.安装pycuda
pycuda安装官网
或者pip3 install pycuda

将lib复制到/usr/lib64/中,sudo cp TensorRT-5.XXX/targets/x86_64-linux-gnu/lib/lib* /usr/lib64/

成功安装了TensorRT,版本信息如下:
GPU:RTX2080Ti
显卡驱动:NVIDIA-Linux-x86_64-440.36.run
Cuda:cuda_10.0.130_410.48_linux.run
Cudnn:cudnn-10.0-linux-x64-v7.6.5.32.tgz
Tensorflow:tensorflow_gpu-1.14.0-cp36-cp36m-manylinux1_x86_64.whl
TensorRT:TensorRT-6.0.1.5.CentOS-7.6.x86_64-gnu.cuda-10.0.cudnn7.6.tar.gz

你可能感兴趣的:(其他)