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安装时,会提示没有适合的版本可以安装。