Jetson Xavier NX で実現するEdge AI (Pytorch, torchvision・pillow編)

はじめに

　前回までは必要最低限のTool群をインストールしました。今回は機械学習に必要なPytorchとtorchvisionのインストールをしていきます。GUIは全く必要ないので今回からはRemoteからSSHで接続しJetson Xavier NX(以下NXとします)を操作していきます。まだSSHで接続できる状態になっていない方は以下のリンクより設定お願いいたします。

Pytorchのインストール

　NX上でPytorchを使えるようにする方法は大きく分けて２つあります。

①NVIDIAの公式Pageよりインストール済みdockerイメージをDLする
②自分でLocalにインストールする
　
　今回は②の方法です。自分で実験した感じではpyenvで仮想環境を構築し、Python及びpipインストール後に、pipよりpytorchをインストールすることも可能ですが、GPUが上手く認識しなかったため何か今回は諦めました。誰か良い方法を知っていたら教えてください。
　なので今回はLocalに仮想環境を構築せず直でインストールする方法となります。初心者には扱いやすい方法かと思います。
　公式Pageに飛ぶとインストール方法が記載されているのでこれに従います。Pytorchは最新の1.8.0があります。