หลังจากสร้าง image disc สำหรับ raspberry pi โดยใช้ Raspberry Pi Imager เรียบร้อยแล้ว (link)
ให้ทำการ boot raspi เข้า Desktop และเปิด terminal

ตรวจสอบเวอร์ชันของ raspi

uname -a

สังเกตเวอร์ชันของ pi เป็น “armv7l GNU/Linux”

ติดตั้ง dependencies

sudo apt install libopenblas-dev libblas-dev m4 cmake cython python3-dev python3-yaml python3-setuptools

จากนั้น set env-var สำหรับการ build pytorch ดังนี้

export NO_CUDA=1
export NO_DISTRIBUTED=1
export NO_MKLDNN=1 
export BUILD_TEST=0
export MAX_JOBS=4

Compile pytorch จาก github src

git clone --branch v1.7.0 https://github.com/pytorch/pytorch pytorch
cd pytorch
git submodule update --init --recursive
python3 ./setup.py bdist_wheel  #ใช้เวลา build ~120 นาที

file *.whl จะถูกสร้างไปที่ ./dist/

ติดตั้ง PyTorch

cd dist/
pip3 install Cython numpy torch-1.7.0a0-cp37-cp37m-linux_armv7l.whl

ทดสอบการติดตั้ง PyTorch ว่าสามารถใช้งานได้

python3
import torch
torch.__version__

x=torch.rand(5,4)
print(x)

หลังจากติดตั้ง Jetson Nano image เรียบร้อยแล้ว (ดูวิธีติดตั้งได้ที่)

ตรวจสอบ Nano version (Jetpack)

sudo su
cd
pip3 --version
pip3 install jetson-stats
jtop
exit #ออกจาก root user
sudo jtop

แถบด้านบนแสดงเวอร์ชัน Jetpack=4.4.1 กดปุ่ม q เพื่อออกจาก jtop

ติดตั้ง PyTorch version 1.9

ดูรายละเอียดเวอร์ชันที่เข้ากันได้กับ Jetpack

wget https://nvidia.box.com/shared/static/h1z9sw4bb1ybi0rm3tu8qdj8hs05ljbm.whl -O  torch-1.9.0-cp36-cp36m-linux_aarch64.whl
sudo apt-get update
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev 
pip3 install Cython
pip3 install numpy torch-1.9.0-cp36-cp36m-linux_aarch64.whl

ติดตั้ง touchvision version 0.10

git clone --branch v0.10.0 https://github.com/pytorch/vision torchvision
cd torchvision
sudo apt-get install libpng-dev libfreetype6-dev
pip3 uninstall pillow
pip3 install --no-cache-dir matplotlib pillow
sudo python3 setup.py install
cd ..

ตรวจสอบว่าติดตั้ง pytorch & torchvision ได้สมบูรณ์

pip3 list | grep torch