Skip to main content

reComputer Jetson用PyTorchのインストール

このwikiでは、お使いのreComputerにJetPackバージョンに対応したNVIDIA最適化PyTorchをインストールする方法を説明します。また、仮想環境（conda）を使用して開発とデプロイメント用に複数のPyTorchバージョンを管理する方法についても説明します。

前提条件​

  • JetPackがインストールされたreComputer
  • インターネット接続

reComputer用PyTorchのインストール​

ここでは、一般的に使用されるJetpack 5とJetpack 6の2つのバージョンでCUDAサポート付きPyTorchをインストールする方法を紹介します。

reComputerでは、Jetpack 5の一般的に使用されるバージョンは5.1.3です。ここでは、これを使用してPyTorchのインストール方法を紹介します。

ステップ 1. システムを更新し、依存関係をインストールします：

sudo apt-get -y update
sudo apt-get install -y python3-pip libopenblas-dev

ステップ 2. 公式PyTorchホイールをダウンロードします（例：JetPack 5.1.3、Python 3.8用のtorch-2.1.0）：

wget https://developer.download.nvidia.com/compute/redist/jp/v512/pytorch/torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl
pip install torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl
note

Jetpack 5.1.3をサポートするより新しいバージョンのtorchが必要な場合は、こちらからコンパイル済みのwheelファイルをダウンロードしてください（python3.8用のtorch-2.2）。

ステップ 3. インストールを確認します：

import torch
print(torch.cuda.is_available())
print(torch.__version__)

Conda を使用した複数の PyTorch バージョン​

Miniconda は Conda パッケージマネージャーの軽量インストーラーで、Python 環境とパッケージを迅速に作成・管理するための最小限の環境を提供します。

ステップ 1. Miniconda をインストール：

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-aarch64.sh
bash Miniconda3-latest-Linux-aarch64.sh
info
  • 指示に従って「yes」と入力し、利用規約に同意してください。
  • インストールパスはデフォルト設定（~/miniconda3など）の使用を推奨します。
  • 最後に.bashrcに自動追加するかどうかの確認が表示されます。「yes」と入力することを推奨します。

ステップ 2. condaを初期化して設定します：

source ~/.bashrc
# Verify if conda is functioning properly
conda --version
# Disable automatic entry into the conda base environment
conda config --set auto_activate_base false
# Close and reopen the terminal. By default, it will not enter the base environment.

ステップ3. 新しい環境を作成し、特定のPyTorchホイールをインストールします：

conda create -n torch_2.0 python=3.8
conda activate torch_2.0

ステップ 4. 前のセクションで示されているように、PyTorchのwheelファイルをダウンロードしてインストールします。

info

詳細については、NVIDIA PyTorchインストールガイドを参照してください

リソース​

技術サポート & 製品ディスカッション​

弊社製品をお選びいただき、ありがとうございます！弊社では、お客様の製品体験が可能な限りスムーズになるよう、さまざまなサポートを提供しています。異なる好みやニーズに対応するため、複数のコミュニケーションチャネルを提供しています。

Loading Comments...