unsurv.org

exposing surveillance

Trainiere deine Objekterkennung für das coral dev board mit einer NVIDIA GPU

de / en

14 Apr, 2021

Hier eine kleine Anleitung um z.B Tensorflow Objekterkennungen für das coral dev board(mini) per GPU zu trainieren:

Folge zunächst dieser Anleitung um die NVIDIA Grafikarte an die Tensorflow Dockercontainer weiterzugeben. Stoppe hier bevor du den Dockercontainer baust https://coral.ai/docs/edgetpu/retrain-detection/#set-up-the-docker-container

In ".../tutorials/docker/object_detection/Dockerfile" ändere Folgende Linie

FROM tensorflow/tensorflow:1.12.0-rc2-devel

zu

FROM tensorflow/tensorflow:1.12.0-rc2-devel-gpu

Folge der Anleitung weiter und benutze "--gpus all" wenn du den Dockercontainer startest. "nvidia-smi" sollte jetzt im Docker in der Kommandozeile verfügbar sein. Das Kommando zum starten fängt so an:

sudo docker run --name edgetpu-detect --gpus all --rm -it --privileged -p 6006:6006

Beende die offizielle Anleitung und trainiere deine Modelle mit deiner GPU. That's it, have fun.