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

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值