Linux 安装TensorRT、Pycuda

Linux 安装TensorRT、Pycuda

文章目录

  • Linux 安装TensorRT、Pycuda
  • 前言
  • 一、下载cuDNN和TensorRt
  • 二、解压与添加路径
  • 三、安装tensort python版
  • 四、安装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__)

你可能感兴趣的:(linux,ubuntu,python)