１．はじめに
前回の記事でRPiにTensorflowLiteをインストールしたのですが、Pytorchにも興味があるのでPytorchもインストールしてみました。一部の依存ライブラリの最新版がRaspiには対応していないようなので Pytorchのバージョンは1.5を入れています。
ついでに画像処理用と音声処理用のライブラリ(torchvisionとtorchaudio)も併せてインストールしました。
PytorchもTensolflowと同じくエッヂAI用に軽量なPytorch mobileというバージョンがある様ですが、Raspiに入れている例がみあたらなかったのでフルサイズのPytorchを入れています。
２．早速入れてみる
いろいろ試行錯誤してなんとかなったのですが、こちらのサイトにほぼ同じ内容が説明されてました。
蛇足もいいとこですが、一応私の実行した方法覚書
関係するモジュール関係をインストール。
インストール環境は、前回TensolflowLiteを入れた環境に追加インストールです。
$ sudo apt install libopenblas-dev libblas-dev m4 cmake cython python3-dev python3-yaml python3-setuptools
$ sudo apt-get install libopenblas-base libopenmpi-dev
$ sudo apt-get install libopencv-highgui-dev
RPi用のwheelらしきリポジトリをclone
$ git clone https://github.com/Ben-Faessler/Python3-Wheels.git
$ cd Python3-Wheels/pytorch
$ sudo pip3 install torch-1.5.0a0+4ff3872-cp37-cp37m-linux_armv7l.whl
$ cd ../torchvision
$ sudo pip3 install torchvision-0.6.0a0+b68adcf-cp37-cp37m-linux_armv7l.whl
$ cd ../torchaudio
$ sudo pip3 install torchaudio-0.5.0a0+09494ea-cp37-cp37m-linux_armv7l.whl
これでpython3を立上げてimport torchでエラーが出なければOK