Jetson Xavier NX で実現するEdge AI (Pytorch-lightning編)

はじめに

　前回で機械学習に必要なTool群はインストールできました。今回はより快適に機械学習をするのに必要な追加のTool群をインストールしておきます。まだ必要なTool群をインストールされていない方はこちらをご参照ください。

Pytorch-lightningとは

　今回インストールするToolはPytorch-lightningです。これはPytorchのラッパーで、Pytorchでの機械学習開発を助けてくれる非常に優れたラッパーです。なので必要最低限のCodeを書けば機械学習ができるようになります。
　また、ラッパーなので基本構成が整っていれば、Mac上、Win上及びColab上でも基本どの環境に持っていっても共通して使えます。なので移植性という意味でも素晴らしいものです。もっと詳しく知りたい方は以下の公式をご参照ください。