redhat-et / ray-on-the-edge-demo

A proof of concept for machine learning workflows using the Ray stack in the data center and at the edge.
Apache License 2.0
0 stars 2 forks source link

[SPIKE] Confirm which ML Frameworks are supported on Jetson + MicroShift #2

Open MichaelClifford opened 2 years ago

MichaelClifford commented 2 years ago

Before proceeding to building and training our model we need to confirm which ML Frameworks (Tensorflow, Pytorch) are supported for our edge devices.

Acceptance Criteria:

MichaelClifford commented 2 years ago

Looks like there is a path from either Tensorflow or Pytorch to TensorRT through Onnx.

MichaelClifford commented 2 years ago

@cmagina Can this be closed now? We've confirmed that pytroch will work on our devices, right?