LoginSignup
1

More than 1 year has passed since last update.

posted at

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

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
What you can do with signing up
1