Ubuntu20.04安装TensorRT

Ubuntu20.04安装TensorRT

一、安装Cuda和Cudnn

注意安装英伟达驱动时不要添加-no-opengl-files这个选项

Ubuntu20.04+RTX3060+Cuda11.1+Pytorch1.9.0 服务器配置

二、安装TensorRT

首先要注意安装TensorRT的版本要与cuda版本和onnxruntime版本对应。

其次下载TensorRT .tar版本 。

安装教程CSDN。

注意在配置环境时就不要看教程了。在配置好环境变量后,一定要再次激活当前的环境。

比如我想把TensoRT安装到pytorch这个我自己创建的环境中,在配置好环境变量后,

一定要再次激活pytorch的环境进行下一步操作。

其他看教程就行

我下载的TensorRT的目录在/home/wyqq/Downloads/下。所以配置环境变量要修改下。

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/wyqq/Downloads/TensorRT-8.0.1.6/lib

你可能感兴趣的:(pytorch,人工智能,ubuntu,深度学习)