TensorRT学习笔记1 - 环境配置
生活随笔
收集整理的這篇文章主要介紹了
TensorRT学习笔记1 - 环境配置
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
目錄
- 版本選擇
- 安裝過程
版本選擇
? ? ? ?ubuntu 16.04LTS
? ? ? ?tensorrt 5.1.5
? ? ? ?cuda 9.0.176
? ? ? ?cudnn 7.5.0
? ? ? ?python 3.5.2
? ? ? ?gcc 5.4.0
安裝過程
? ? ? ?首先,cuda、cudnn、python和gcc的版本一定要和選擇的TensorRT版本對應,可在此鏈接查看。下圖為TensorRT5.1.5所需的版本。
? ? ? ?TensorRT的下載鏈接。需要登錄NVIDIA帳號。我下載的是<TensorRT 5.1.5.0 GA for Ubuntu 16.04 and CUDA 9.0 tar package>。不下deb文件的原因是:使用deb安裝TensorRT必須滿足cuda和cudnn也是使用deb安裝的。
? ? ? ?官方安裝教程在這里。我使用的是4.3中的方法進行安裝,簡單易懂。
總結
以上是生活随笔為你收集整理的TensorRT学习笔记1 - 环境配置的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 360企业安全2019暑期实习算法岗笔试
- 下一篇: TensorRT学习笔记2 - 基础知识