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に掲載されている手順にを参考に、インストールします。
ただし、後述する①②に注意してください。
①最新のバージョンを使用できない
最新のバージョンはインストールできませんでした。(ビルドエラーになりました)
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
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のバージョンを表示