Can I use the hailo8 board on the pi5 with pytorch to train a network or just to use it? Are there drivers so I can use the board to hardware accellerate pytorch?
Welcome to the Hailo Community!
The Hailo-8 is an accelerator to run inference not training.
(1) You can develop and train a network using PyTorch. Then the model can be exported to ONNX.
(2) With the help of the Hailo Dataflow Compiler the ONNX file can then be converted in a few steps into a HEF (Hailo Executable File).
(3) That HEF file will be loaded by the HailoRT runtime into the Hailo-8 to turn it into an accelerator for that network.
I now have a neural network (U-Net) built with pytorch, also trained on the pc side, what I should do to get it running on Raspberry PI 5 and call hailo8 for acceleration