PytorchによるDeepLearningの動作環境（PyTorch＋Torchvison＋OpenCV）をRaspberryPi4に構築する方法

書籍「つくりながら学ぶ! PyTorchによる発展ディープラーニング」を参考にして作成したDeepLearningのプログラムをRaspberryPi4上で動作させるために、PyTorch、Torchvision、OpenCVの3つをインストールする必要がありました。

この記事には、これら3つのモジュールのインストール手順のメモを掲載します。
ただし、Raspbian上で実行したものであり、Raspberry Pi OS上で実行した場合に動作するかを確かめていない点、ご容赦ください。

環境

2020/4/3時点で最新のRaspbianを使いました。
（このメモを作成時はRaspberry Pi OSが公開されていませんでした・・・）

OS：Raspbian Buster(Release date:2020-02-13)
Python：3.7.3

インストール

1. Pytorchをインストール

GitHubに掲載されている手順にを参考に、インストールします。
ただし、後述する①②に注意してください。

PyTorchのインストール手順

①最新のバージョンを使用できない

最新のバージョンはインストールできませんでした。（ビルドエラーになりました）
Optionally・・・の箇所に記載されている

git checkout v1.0.1

を実施すると、無事にインストールできました。
git checkout XXXのXXX部分は、GitHubのBranch名称に対応しています。

②PyTorchとTorchvisionのバージョンの組み合わせ
PyTorchとTorchvisionのバージョンの組み合わせは、以下のリンク先のページから選択する必要があります。
PyTorchとTorchVisionの組み合わせ
今回は、Pytorch：v1.0.1、Torchvision：v0.2.2の組み合わせでインストールすることにしました。

2. Torchvisionをインストール

Torchvisionをインストールします。
root で以下を実行してください。

git clone https://github.com/pytorch/vision.git
cd vision
git checkout v0.2.2_branch
python3 setup.py setup build
python3 setup.py setup install

参考ページ：https://gist.github.com/akaanirban/621e63237e63bb169126b537d7a1d979

3. OpenCVをインストール

次のコマンドを実行すると、OpenCVがインストールされます。

sudo pip3 install opencv-python

動作確認

ターミナルからpythonを起動し、次のコードを実行してください。
PyTorch、Torchvision、OpenCVのバージョンがそれぞれ表示されればOKです。

import torch
import torchvision
import cv2
torch.__version__  # Pytorchのバージョンを表示
torchvision.__version__ # Torchvisionのバージョンを表示
cv2.__version__ # OpenCVのバージョンを表示