forked from pjreddie/darknet
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Dockerfile
46 lines (35 loc) · 876 Bytes
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04
MAINTAINER Loreto Parisi loretoparisi@gmail.com
# working directory
WORKDIR /darknet
# set noninteractive for TZ data install. Need to pass TZ as env parameter to image
ARG DEBIAN_FRONTEND=noninteractive
# install
RUN \
apt-get update && apt-get install -y \
autoconf \
automake \
libtool \
build-essential \
git # \
# libopencv-dev
# addons
RUN \
apt-get install -y \
wget
# build repo
RUN \
git clone https://github.com/Entr0py86/darknet
WORKDIR darknet
RUN \
sed -i 's/GPU=.*/GPU=1/' Makefile && \
# sed -i 's/OPENCV=.*/OPENCV=1/' Makefile && \
make
# download weights full (accurate most) and tiny (faster , less accurate) models
# darknet rnns
RUN \
wget https://pjreddie.com/media/files/yolov3.weights >/dev/null 2>&1
# test nvidia docker
CMD nvidia-smi -q
# defaults command
CMD ["bash"]