项目作者: JulienCheny

项目描述 :
run darknet in docker with cuda 10
高级语言: Dockerfile
项目地址: git://github.com/JulienCheny/darknet-docker.git
创建时间: 2019-03-06T15:26:56Z
项目社区:https://github.com/JulienCheny/darknet-docker

开源协议:

下载


darknet-docker

Required host dependancies

Install dependancies command caryall

Driver nvidia

  1. sudo ubuntu-drivers autoinstall

Required packages

  1. sudo apt-get install build-essential cmake git apt-transport-https ca-certificates curl software-properties-common

Cuda

  1. sudo dpkg -i cuda-repo-ubuntu1810_10.1.105-1_amd64.deb
  2. sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1810/x86_64/7fa2af80.pub
  3. sudo apt-get update
  4. sudo apt-get install cuda

Docker

  1. curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
  2. export LSB_ETC_LSB_RELEASE=/etc/upstream-release/lsb-release
  3. V=$(lsb_release -cs)
  4. sudo add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu ${V} stable"
  5. sudo apt update -y
  6. sudo apt install -y docker-ce

Docker-compose

  1. curl -L https://github.com/docker/compose/releases/download/1.24.0-rc1/docker-compose-`uname -s`-`uname -m` -o /usr/local/bin/docker-compose
  2. chmod +x /usr/local/bin/docker-compose

Nvidia-docker-v2

  1. curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
  2. curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu18.04/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
  3. sudo apt-get update
  4. sudo apt-get install nvidia-docker2
  5. sudo pkill -SIGHUP dockerd

Docker-compose commands

  1. docker-compose up -d
  2. docker-compose exec darknet bash
  3. docker-compose exec darknet ./darknet detect cfg/yolov3-tiny.cfg yolov3-tiny.weights data/dog.jpg