LoginSignup
1

More than 1 year has passed since last update.

PytorchによるDeepLearningの動作環境(PyTorch+Torchvison+OpenCV)をRaspberryPi4に構築する方法

Posted at

PytorchによるDeepLearningの動作環境(PyTorch+Torchvison+OpenCV)をRaspberryPi4に構築する方法

書籍「つくりながら学ぶ! PyTorchによる発展ディープラーニング」を参考にして作成したDeepLearningのプログラムをRaspberryPi4上で動作させるために、PyTorchTorchvisionOpenCVの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

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のバージョンを表示

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
1