darknet-docker

This repo contains a collection of Docker images that can be used to run darknet (YOLO) on different devices, primarily the NVIDIA Jetson TX2.

Running on Jetson TX2

Because nvidia-docker was not availabe for the Jetson TX2 at the time this was written, I've included a similar wrapper script under jetson-tx2/darknet-docker which can be used in place of docker run on the Jetson TX2 like so:

./darknet-docker run --rm -it jcjimenez/darknet-docker:jetson-tx2-latest

Running on a GPU machine

Make sure you have nvidia-docker installed to run:

nvidia-docker run --rm -it jcjimenez/darknet-docker:gpu-latest

Running on a plain CPU machine

Since this is a CPU-based image, you can just use the regular docker command:

docker run --rm -it jcjimenez/darknet-docker:cpu-latest