ベリーパイにpytorchをインストール(python 3.7)

2029 ワード

pytorchソースからのコンパイルインストール


前提条件sdは16 Gbまたは32 Gbであることが好ましく、システムをインストールした後、システムを更新する.

スワップ領域のサイズを2 Gに増やします


まずコマンドでスワップ領域設定ファイルを開きます
sudo vim /etc/dphys-swapfile

次に定数CONF_を見つけますSWALSEIZEは、後のデフォルトの100を2048に変更します.次に、次のコマンドを実行してファイル交換システムをアクティブにします.
sudo /etc/init.d/dphys-swapfile stop
sudo /etc/init.d/dphys-swapfile start

依存パッケージのインストール
sudo apt-get install libopenblas-dev cython3 libatlas-dev m4 libblas-dev cmake

仮想環境の作成


このときvenvで必要な仮想環境を作成し、pytorchを他のpythonパッケージから分離することができます.次のコマンドを実行できます.
python3 -m venv pytorch
cd pytorch/bin
source activate

このとき、コマンドウィンドウのプロンプトの前に(pytorch)を見ることができます.これは、仮想環境に入ったことを意味します.

pytorchソースコードのダウンロード


この時gitでソースコードをダウンロードできます.gitを使用してダウンロードするプロセスは遅いかもしれませんが、ローカルでダウンロードしてベリーパイに転送することもできます.
git clone --recursive https://github.com/pytorch/pytorch

pytorchインストールディレクトリを開くには、次の手順に従います.
cd pytorch

v 1などのインストールしたいバージョンに切り替えます.0.1:
git checkout tags/v1.0.1 -b build
git submodule update --init --recursive

コンパイルpytorch


このときpytorchのコンパイルを開始することができます.しかし、pytorchをコンパイルする前に、依存するpythonパッケージをインストールする必要があります.依存パッケージのインストール:
pip install -r requirements.txt


環境変数の設定


このステップは非常に重要です.環境変数の設定が正しくない場合、コンパイルのプロセスが間違っています.例:
Failed to run 'bash tools/build_pytorch_libs.sh --use-cuda --use-nnpack --use mkldnn --use qnnpack caffe2'

私たちはexportコマンドを使用して、必要な環境変数を設定することができます.例えば、ベリーパイはcudaをサポートしていないので、cudaサポートを削除します.
export NO_CUDA=1
export NO_DISTRIBUTED=1
export NO_MKLDNN=1
export NO_NNPACK=1
export NO_QNNPACK=1

次に、コマンドを使用します.
python setup build

pytorchをコンパイルするには、数時間かかります.コンパイルが完了すると、コマンドでインストールできます.
python setup install

これでpytorchはインストールを完了します.