当运行tensorrt中的sample时,发现居然需要tensorflow是1.15...
1.创建新的conda环境
conda create -n tensorrt_tf1.15 python=3.6
conda acitvate tensorrt_tf1.15
2.安装cuda、cudnn、tf1.15
conda install cudatoolkit=10.0.130 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
conda install cudnn=7.6.4 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.15.0
3.安装tensorrt7
下载trt7的zip包、解压、export环境、安装对应python3.6的trt
trt7包选择对应cuda版本、系统版本 的下载包(需要注册用户、勾选协议、填写信息之后就可以下载了)
t