YOLOV3 GPU环境配置

DarkNet搭建GPU运行环境

当前环境配置

• ubuntu 16.04
• cuda (已安装)
• cudnn (已安装)

安装步骤

• 获取开源darknet框架
  git clone https://github.com/pjreddie/darknet
• 修改Makefile文件
  • cd darknet/
  • vim Makefile
  • GPU=1
  • CUDNN=1
• 编译
  make
• 获取yolov3权重文件
  wget https://pjreddie.com/media/files/yolov3.weights
• 运行detect demo
  ./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg

运行耗时

GPU使用的是gtx1080ti，只花费了0.03s。