安装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了,选择

  • TensorRT 8.6 EA for Linux x86_64 and CUDA 11.0, 11.1, 11.2, 11.3, 11.4, 11.5, 11.6, 11.7 and 11.8 TAR Package

基本可以支持所有的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的版本号就是安装成功了。

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.hqwc.cn/news/318482.html

如若内容造成侵权/违法违规/事实不符,请联系编程知识网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

38个城市平均薪酬超1万元;丁俊晖6-5绝杀世界第一奥沙利文夺冠;原钉钉副总裁创业杀入AI Agent赛道

投融资 • 原钉钉副总裁创业杀入AI Agent赛道,AI 公司“斑头雁智能科技”获亿元融资• 荣盛石化与沙特阿美拟再相互收购下属公司股权• 斯年智驾完成B轮融资• 欧冶半导体完成A3及A4轮融资• 耀途天使轮项目旗芯微完成数亿元新一轮融资• 野牛王获近千万元天使轮融…

教育机构培训系统小程序功能清单

制作一款适合自己的教育机构培训系统小程序,可以为学员提供更便捷的学习体验,同时提高机构的教学效率。今天将详细介绍如何使用乔拓云平台制作教育机构培训系统小程序。 在浏览器搜索乔拓云,登录到后台,选择教育系统并点击进入。在…

2023年互联网公司年度崩盘报告

B站崩了两次 2023年3月5日晚20:20左右,许多网友表示在使用B站时,手机和电脑端都无法访问视频详情页,且手机端无法查看收藏夹与历史记录。 8月4日晚间,距离上次事故5个月后,又有许多网友反馈B站图片(视频封…

8K自动化测试面试题分享(有答案,非常详细)

关于自动化测试面试,会问到哪些问题呢?给大家简单总结了一下,每一个都是学员反馈过来的企业真题,相信对大家有帮助,最近有面试机会的,快来背一下答案吧 1、你会封装自动化测试框架吗? 这个问得…

软件测试基础篇(超详细整理)

1、软件测试的生命周期 回顾:需求分析、计划、设计、编码、测试、运行维护 软件测试的生命周期: 需求分析→测试计划→ 测试设计→ 测试开发→ 测试执行→ 测试评估 需求分析:分析需求的正确性,合理性。细化需求,得…

TB-C/C++

1.main函数之前之后执行的代码 设置栈指针初始化静态变量和全局变量(.data段内容,已初始化且不为0)赋初值(.bss段内容,未初始化的全局变量和静态变量)传参(argc,argv)atexit() 在…

12.Harbor构建私有镜像仓库

1、阿里云容器镜像服务-个人版 细心的同学可能已经发现,在前面的部署过程中,前面所有的部署步骤中需要的镜像都是从阿里云的镜像仓库中下载。 因为网络原因,有的镜像可能下载比较慢,有点可能下载不了,所以为了加速镜像下载,我都统一将镜像推送到阿里云的镜像仓库(个人…

编译器功能__attribute__介绍和官方资料来源

1、__attribute__简介 __attribute__不是C语言本身的关键字,而是属于编译器扩展的C语言的功能,C Extensions (Using the GNU Compiler Collection (GCC))中可以找到关于attribute的几个章节,Function Attributes【函数属性】、Variable Attr…

基于深度学习的交通标志图像分类识别系统

温馨提示:文末有 CSDN 平台官方提供的学长 QQ 名片 :) 1. 项目简介 本文详细探讨了一基于深度学习的交通标志图像识别系统。采用TensorFlow和Keras框架,利用卷积神经网络(CNN)进行模型训练和预测,并引入VGG16迁移学习…

基于YOLOv7算法的高精度实时海上船只目标检测识别系统(PyTorch+Pyside6+YOLOv7)

摘要:基于YOLOv7算法的高精度实时海上船只目标检测系统可用于日常生活中检测与定位海上船只,此系统可完成对输入图片、视频、文件夹以及摄像头方式的目标检测与识别,同时本系统还支持检测结果可视化与导出。本系统采用YOLOv7目标检测算法来训…

基于华为ENSP模拟器-vlan划分网络

需求 不连外网的内网。需求隔离故障和隔离广播风暴,并要保证网络的连通。 解决方案使用三层交互机,设置vlan用于隔离网络,并在三层交互机为网关保证各个vlan之间的通讯。 实现 使用三层交互机,设置vlan用于隔离网络&#xff0…

全国计算机等级考试| 二级Python | 真题及解析(11)

一、选择题 1.有关循环结构的说法不正确的是( )。 A.循环结构是算法的基本结构之一 B.有的的程序设计中没有循环结构 C.循环结构在程序设计有可能会有嵌套出现 D.在PYTHON 程序设计语言中循环结构一般使用IF语句实现。 2.在Python中要交换变量a和b中的值,应使…