7-1 PyTorch 環境の構築

初期環境設定
➤ JetPackおよびPython関連ツールのインストールについては、「第2章第3節：Jetsonコンポーネント環境の構築」をご参照ください。

sudo apt update && \

sudo apt install nvidia-jetpack -y && \

sudo apt install python3-pip -y

PyTorchをインストールしてテストする

インストール可能なバージョンを確認する

// Install dependencies

sudo apt install libopenblas-dev -y
// Verify available versions

pip3 index versions torch -i https://pypi.jetson-ai-lab.io/jp6/cu126
// Install a specific version (this tutorial uses v2.8.0)

pip3 install torch==2.8.0 -i https://pypi.jetson-ai-lab.io/jp6/cu126
// Test the installation

python3 -c '

import torch

print(f"PyTorch version: {torch.__version__}")

print(f"CUDA available:  {torch.cuda.is_available()}")

print(torch.__config__.show())

'