Jetson Nano 2GBのセットアップ ~Pytorchインストール編~

はじめに

Jetson Nano 2GB とは NVIDIA社によって販売されている小型のエッジコンピューティング用のGPUコンピューティングボードです。これを使って Kaggleの学習をさせるためにセットアップします。今回はPytorchをJetson Nano 2GBにインストールします。
行うことはこちらに書いてあることと同じため公式の方が良い方はこちらをみてください。
ただし、注意点もあるので気をつけてください。

インストールするバージョンについて

2023/2/24現在においてJetson Nano 2GB用の最新JetPack 4.6.1 はPython3.6であるため使用できるPytorch のバージョンは1.10.0までである。そのため、JetPack5.1用のインストール方法を試してもうまくいかないため、こちらに書いてあるインストール方法を試しましょう。

インストール

前回の続きから行います。まだみていない方はこの記事からセットアップしてください。
セットアップしたら以下のコマンドを入力してください。

# install the dependencies (if not already onboard)
sudo apt-get install python3-pip libjpeg-dev libopenblas-dev libopenmpi-dev libomp-dev
sudo -H pip3 install future
sudo pip3 install -U --user wheel mock pillow
sudo -H pip3 install testresources

# above 58.3.0 you get version issues
sudo -H pip3 install setuptools==58.3.0
sudo -H pip3 install Cython

# install gdown to download from Google drive
sudo -H pip3 install gdown

# download the wheel
gdown https://drive.google.com/uc?id=1TqC6_2cwqiYacjoLhLgrZoap6-sVL2sd

# install PyTorch 1.10.0
sudo -H pip3 install torch-1.10.0a0+git36449ea-cp36-cp36m-linux_aarch64.whl

# clean up
rm torch-1.10.0a0+git36449ea-cp36-cp36m-linux_aarch64.whl

エラーが出た場合は以下を実行してもう一度# download the wheelから実行し直しましょう。

最後に以下でクリーンアップしましょう。

sudo apt autoremove