安装tensorrt环境在linux上

在linux上输入命令

bash

cat /etc/os-release

命令查看系统版本

nvidia-smi命令后有内容弹出而没有报错,表明系统中安装了NVIDIA显卡驱动,并且该命令成功地显示了有关NVIDIA GPU的信息。

输入nvcc -V并且看到输出时,这表明您的系统中已经安装了NVIDIA的CUDA工具包,并且该命令成功地显示了CUDA编译器版本的信息

这里是租的ubuntu远程服务器gpu版本,所以nvidia的驱动和cuda都是安装上了。

输入which nvcc 或 where nvcc即可查看cuda的安装路径

可以看到cuda的安装路径是在/usr/local/cuda/bin/nvcc的路径下

除了cuda还需要cudnn

输入指令查看是否存在cudnn

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

弹出以上内容说明cudnn是存在的,这样安装tensorrt的前置工作就完成了,接下来我们进行tensorrt的安装。

这里直接安装tensorrt相对应的版本即可,在官网地址下载:

Log in | NVIDIA Developer

我这里直接下载linux版本 不去区分ubuntu还是centos了,选择

基本可以支持所有的cuda11了。

安装包下载好后之间拖到服务器上。我这里的路径直接拖到/usr/local的路径下。

安装包拖过去后,输入命令完成解压。

tar -xvf /usr/local/TensorRT-8.6.0.12.Linux.x86_64-gnu.cuda-11.8.tar.gz

完成解压,如果没有权限 就在最前面加上sudo

开始解压

因为我们当前路径是在root下,所以解压的文件在root文件夹下

然后我们开始添加环境路径

export LD_LIBRARY_PATH=/root/TensorRT-8.6.0.12/targets/x86_64-linux-gnu/lib:$LD_LIBRARY_PATH

输入echo $LD_LIBRARY_PATH 查看环境路径

可以看到路径已经加载进来了

然后我们使用pip安装

当前已经是cd在了TensorRT-8.6.0.12的路径下了,

我们直接输入指令安装

pip install ./python/tensorrt-8.6.0-cp38-none-linux_x86_64.whl安装

然后输入以下指令

import tensorrt as rt

print(rt.version)

·正确打印出了tensorrt的版本号就是安装成功了。

相关推荐
学编程就要猛14 分钟前
RAG(Retrieval Augmented Generation)
人工智能·spring ai
冬奇Lab1 小时前
一天一个开源项目(第259篇):Modern Web Guidance —— Chrome 团队给 AI 编码 Agent 的「现代 Web 最佳实践」外挂
人工智能·开源
§ꦿCFོ༉2 小时前
嵌入式八股文—C语言入门学习(四)
c语言·开发语言·学习
飞哥数智坊2 小时前
打造自己的 Personal Agent,从抄大肥鱼作业开始
人工智能·deepseek
投图匠2 小时前
十种不同风格的 CTP接口 框架
开发语言·python
对你无可奈何3 小时前
Magpie 实战:把 WorkBuddy、Qoder 的订阅模型接进一个本地池子
人工智能
Csvn3 小时前
每请求到底多少钱?把 Token 成本算成可观测指标(C01)
人工智能·aigc·agent
吴名氏.3 小时前
Java转AI Agent教程(第十三周)混合检索与重排
java·数据库·人工智能·ai·ai agent
2601_967980114 小时前
深入理解词向量算法及其在Python中的实践
python·word2vec·glove·fasttext·词向量