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.