: O. Yuanying

Jetson Orin Nano に PyTorchとtorchvisionをインストール

JetPack 5.10 がインストールされたJetson上のコンテナにPyTorchとtorchvisionをインストールしたい。そのままインストールができなかったため以下を参考にする。

PyTorch のインストール

これはドキュメントを参考に。

$ export TORCH_INSTALL=https://developer.download.nvidia.cn/compute/redist/jp/v51/pytorch/torch-2.0.0a0+8aa34602.nv23.03-cp38-cp38-linux_aarch64.whl
$ python3 -m pip install --no-cache $TORCH_INSTALL

torchvisionのインストール