Jetson Orin での PyTorch 非互換問題
問題 1: GPU が使えない
Jetson 版 PyTorch をインストールします。
チュートリアル: https://docs.nvidia.com/deeplearning/frameworks/install-pytorch-jetson-platform/index.html
バージョンは自分で選べます: https://developer.download.nvidia.com/compute/redist/jp/
ここで使用したバージョン:
torch-2.5.0a0+872d972e41.nv24.08.17622132-cp310-cp310-linux_aarch64.whl
ダウンロード後、実行:
bash
export TORCH_INSTALL=torch-2.5.0a0+872d972e41.nv24.08.17622132-cp310-cp310-linux_aarch64.whl
pip install --no-cache $TORCH_INSTALLテスト:
python
import torchpython
torch.cuda.is_available()問題 2: libcusparseLt エラー
エラー: ImportError: libcusparseLt.so.0: cannot open shared object file: No such file or directory
解決方法:
Jetson Orin プラットフォームが対応する CUDA バージョンと一致する cuSPARSELt ライブラリをインストールします。
ダウンロードリンク:
https://developer.nvidia.com/cusparselt-downloads
問題 3: torchvision がない
解決方法:
対応する torchvision バージョンを手動でインストールします。torch 2.5 → torchvision 0.20.0
python
git clone --branch v0.20.0 https://github.com/pytorch/vision.gitダウンロード後、ビルド:
bash
export BUILD_VERSION=0.20.0
python3 setup.py install --user