Linux 安装TensorRT、Pycuda

Linux 安装TensorRT、Pycuda

---

前言

记录安装TensorRt、pycuda的过程

一、下载cuDNN和TensorRt

1、下载cuDNN
2、下载TensorRt

需要注意下载对应cuda版本。

二、解压与添加路径

1、解压

tar -xzvf TensorRT-7.0.0.11.Ubuntu-18.04.x86_64-gnu.cuda-10.0.cudnn7.6.tar.gz
tar -xzvf cudnn-10.0-linux-x64-v7.6.5.32.tgz

2、添加路径

export LD_LIBRARY_PATH=/…/cuda/lib64:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/…/TensorRT-7.0.0.11/targets/x86_64-linux-gnu/lib:$LD_LIBRARY_PATH

3、查看添加的路径

echo $LD_LIBRARY_PATH

三、安装tensort python版

cd TensorRT-7.0.0.11/
cd python
pip install tensorrt-7.0.0.11-cp37-none-linux_x86_64.whl

四、安装pycuda

pip install pycuda -i https://pypi.tuna.tsinghua.edu.cn/simple

五、验证

import tensorrt as rt
import pycuda
print(rt.__version__)