1.下载和cuda版本对应的Tensor安装包。
2.将当前的Tensor添加到环境变量,比如:
sudo vi ~/.bashrc
在最后添加一行
export LD_LIBRARY_PATH=/路径/TensorRT-8_0_1_6/lib:$LD_LIBRARY_PATH
再运行命令使其生效
source ~/.bashrc
3.先激活conda环境,进入tensorrt压缩之后的文件夹,然后进入里面的python文件夹,在conda环境下安装tensorrt的python包。
(4、5步骤可选安装)
4.进入uff文件夹,在conda环境下安装安装 uff 包,比如:
pip install uff-xxxx-any.whl
测试是否安装成功,比如:
which convert-to-uff
(看是否输出安装路径)
5.进入graphsurgen文件夹,在conda环境下安装安装 包,比如:
pip install graphsurgen-xxxx-any.whl