Mobilenet-SSD训练环境搭建和训练自己的数据

最新推荐文章于 2024-08-05 17:44:35 发布

别打名名

最新推荐文章于 2024-08-05 17:44:35 发布

阅读量6.4k

点赞数 2

本文链接：https://blog.csdn.net/FreeApe/article/details/79306011

版权

本文档详细介绍了在Ubuntu14.04上搭建Mobilenet-SSD训练环境的过程，包括安装CUDA 8.0、cudnn、Caffe及其依赖库。此外，还解析了训练超参数，如base_lr、lr_policy等，并指导如何制作LMDB数据集，最后提供训练脚本和部署模型的方法。

摘要由CSDN通过智能技术生成

环境
准备
编译Caffe
训练超参数详解
训练自己的数据
- - 制作LMDB文件
  - 训练

环境

Ubuntu14.04
GTX1070
gcc4.9.x(默认4.8，需升级)

准备

显卡驱动安装
- http://www.nvidia.com/download/driverResults.aspx/130646/en-us (NVIDIA-Linux-x86_64-390.25.run)

# 禁用Ubuntu自带的显卡驱动
sudo vim /etc/modprobe.d/blacklist.conf
    blacklist nouveau option nouveau modeset=0
sudo update-initramfs -u
重启电脑
# 切换到文字界面(Ctrl+Alt+F1~F6)
sudo service lightdm stop
sudo ./NVIDIA-Linux-x86_64-375.66.run
再次重启电脑
# 验证显卡驱动是否安装成功
nvidia-smi

CUDA下载和安装
- https://developer.nvidia.com/cuda-downloads （2018.2.8链接最新是9.0的，下载8.0 https://developer.nvidia.com/cuda-80-ga2-download-archive Linux x86_64 Ubuntu 14.04 runfile(local)）

# 注意你是否已经安装过 nvidia 显卡驱动(即第一步)，若无法保证已安装的 nvidia 显卡驱动一定正确，那就卸载掉之前安装的 nvidia 显卡驱动，
# 然后开始安装 CUDA 8.0；若可以保证已安装正确的 nvidia 显卡驱动，则直接开始安装 CUDA 8.0，在安装过程中选择不再安装 nvidia 显卡驱动
# 安装CUDA
chmod u+x cudaxxxx
sudo sh cuda_8.0.61_375.26_linux.run
    提示是否安装nvidia375驱动(否)，其他选项根据提示选择和设置。
# 环境变量配置
sudo vim  /etc/profile
    export PATH=/usr/local/cuda-8.0/bin:$PATH
    export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

source /etc/profile

# 验证是否安装成功
nvcc –V
# 测试CUDA sample
cd /usr/local/cuda-7.5/samples/1_Utilities/deviceQuery
make
sudo ./deviceQuery