本文详细介绍了如何在Ubuntu 16.04上配置环境，包括Anaconda、CUDA、cuDNN和TensorFlow-GPU的安装，以及DeepLabv3模型的训练和测试。首先，通过安装Anaconda和指定版本的Python、TensorFlow-GPU来搭建基础环境。接着，下载Cityscapes数据集和预训练模型，转换数据集格式为tfrecord。然后，使用训练脚本进行训练，并保存权重。最后，进行模型验证和结果可视化。
0x00 操作环境
OS: Ubuntu 16.04 LTS CPU: Intel® Core™ i7-4790K GPU: GeForce GTX 1080/PCIe/SSE2 Nvidia Driver Version: 384.130 RAM: 32 GB
Anaconda: 4.6.11 CUDA: 9.0 cuDNN: 7.3.1 python: 3.6.8 tensorflow-gpu: 1.13.1
本文操作路径基于 /home/ai，使用 ~/ 代替
0x01 环境配置
安装 Anaconda
首先下载安装脚本并赋予执行权限
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2019.03-Linux-x86_64.sh
chmod +x Anaconda3-2019.03-Linux-x86_64.sh
运行安装脚本
./Anaconda3-2019.03-Linux-x86_64.sh
按照提示输入安装信息，建议安装在 /usr/local/anaconda3 目录下，方便管理
安装 python 3.6
最新版的 Anaconda 默认在 base 环境安装 python 3.7，导致很多框架不支持，这里换成 3.6
conda install python=3.6
安装 tensorflow-gpu
conda install tensorflow-gpu=1.13.0
安装 CUDA
conda install cudatoolkit=9.0
0x02 配置 DeepLab v3
clone 源码
git clone https://github.com/tensorflow/models.git
下载数据集
百度云链接 提取码： 7jgc
在 research/deeplab/datasets/dataset 目录下新建 dataset 文件夹，并将下载的数据集解压至 model-master/research/deeplab/datasets/dataset
mkdir model-master/research/deeplab/datasets/dataset
unzip cityscapes.zip -d model-master/research/deeplab/datasets/dataset
解压 gtFine 文件
cd model-master/res
最低0.47元/天 解锁文章
896
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
