IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> 系统运维 -> ubuntu TensorRT源码编译 -> 正文阅读

[系统运维]ubuntu TensorRT源码编译

1.目的:TensorRT_v6.0.1
2.编译环境准备:

(1) cuda-10.0 + cuDNN-7.6
    CMake >= v3.13若本机CMake<v3.13,则可以搭建conda 环境,在conda 环境里安装CMake=3.15,并记录路径如:/media/data1/3rdtool/anaconda3/envs/libtorchcpp/bin/cmake  
(2)默认 cuda-10.0 和 cuDNN-7.6安装的比较熟悉了;

3.下载TensorRT-6.0.1.5,解压路径:/xx/data11/3rdparty/TensorRT-6.0.1.5/lib

4.在CMakeLists.txt中project()上方添加:

set(CMAKE_CUDA_COMPILER  /usr/local/cuda-10.0/bin/nvcc)
否则会报错:
CMake cannot find CUDA: "Could not find cmake module file: CMakeDetermineCUDACompiler.cmake"

5.源码clone

 git clone -b master https://github.com/nvidia/TensorRT TensorRT
 cd TensorRT

 切换分支:
 git tag
 git checkout  v6.0.1
 
 git submodule update --init --recursive

6.习惯使用clion打开代码:

(1) 在file->setting->build->cmake,在cmake options中写入:
      -DTRT_LIB_DIR=/xx/data11/3rdparty/TensorRT-6.0.1.5/lib
      
(2) 在file->setting->build->Toolchains, 在Cmake选择:
     /media/data1/3rdtool/anaconda3/envs/libtorchcpp/bin/cmake
     
(3)编译,待编译完成把*.so库放入新建文件夹lib中
    /xx /TensorRT/lib

(4)(可选)添加环境变量:Run/Debug Configuration ----->Environment variables:
    LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:/xx/TensorRT/lib

7.然后可以快乐的build运行了。

  系统运维 最新文章
配置小型公司网络WLAN基本业务(AC通过三层
如何在交付运维过程中建立风险底线意识,提
快速传输大文件,怎么通过网络传大文件给对
从游戏服务端角度分析移动同步(状态同步)
MySQL使用MyCat实现分库分表
如何用DWDM射频光纤技术实现200公里外的站点
国内顺畅下载k8s.gcr.io的镜像
自动化测试appium
ctfshow ssrf
Linux操作系统学习之实用指令(Centos7/8均
上一篇文章      下一篇文章      查看所有文章
加:2022-03-17 22:37:43  更:2022-03-17 22:41:23 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁

360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年11日历 -2024/11/16 1:28:57-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码