Ubuntu18.04 安装配置BisNetv2-Tensorflow踩坑记录

Ubuntu18.04 安装配置BisNetv2-Tensorflow踩坑记录

欢迎关注 —> 我的博客原文

尝试了一段时间PaddleSeg，被kitti和cityscapes的数据集格式搞到头晕，我又是DL的纯小白，整个配置文件都云里雾里。正好最近在相关的公众号上看到有新的实时语义分割模型，换个口味试试。

参考：Unofficial tensorflow implementation of real-time scene image segmentation model “BiSeNet V2: Bilateral Network with Guided Aggregation for Real-time Semantic Segmentation”

准备工作

Github推荐的测试环境：

• Ubuntu 16.04(x64)
• python 3.5
• cuda-9.0
• cudnn-7.0
• GTX-1070
• tensorflow-gpu 1.12.0

我自己这边的为了PaddleSeg配置的环境：

• Ubuntu 18.04(x64)
• python 3.6.9
• cuda-10.1
• cudnn-7.6.5
• GTX-1070

首先在这个库的requirements列表里tensorrt这个tensorflow的组件对依赖项的版本有比较苛刻的要求。requirements里要求7.0.0.11版本，参考TensorRT官方文档，需要以下依赖：

• 9.0,10.0,10.2的CUDA Toolkit
• TensorFlow 1.14.0（以上）

之前安装的CUDA10.1并不支持tensorrt，因此在使用pip安装时，会提示没有适合的版本可以安装。