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

运行耗时

YOLOV3 GPU环境配置_第1张图片
GPU使用的是gtx1080ti,只花费了0.03s。

你可能感兴趣的:(深度学习)