一、本文档所使用的软件版本
Python版本:3.7.11
pytorch版本: 1.8.0
CUDA版本:11.1
二、自动安装
在Linux下安装tensorrt包时使用
pip install tensorrt
pip install nvidia-pyindex
pip install nvidia-tensorrt
运行上述命令之后就可以安装成功,不支持Windows环境。
三、手动安装
3.1 百度网盘获取需要的软件包
自动安装有时候版本太高,而我们又需要特定的安装版本。这时就需要我们进行手动安装。手动安装的过程略显麻烦,主要是要到官网下载适合的版本。<