- uff模块安装
根据机器属性下载Tensorrt安装包(对应路径下有tensorrt的安装说明)。解压文件后找到uff安装文件的路径,使用如下命令安装。
pip install TensorRT-4.0.1.6/uff/uff-0.4.0-py2.py3-none-any.whl
- graphsurgeon模块安装
pip install TensorRT-4.0.1.6/graphsurgeon/graphsurgeon-0.2.0-py2.py3-none-any.whl
参考文献: