PyTorch は、TensorFlow と同様に、NVIDIA CUDA ライブラリを介して NVIDIA GPU を使用してネイティブに AI/ML アプリケーションを高速化できます。
この記事では、NVIDIA GPU/CUDA アクセラレーションをサポートする PyTorch を Debian 12「Bookworm」にインストールする方法を説明します。
内容のトピック:
- Debian 12 への NVIDIA GPU ドライバーのインストール
- Debian 12 への NVIDIA CUDA のインストール
- Debian 12 への Python 3 PIP および Python 3 仮想環境 (venv) のインストール
- PyTorch 用の Python 3 仮想環境の作成
- Python 3 PyTorch 仮想環境での Python 3 PIP の最新バージョンへのアップグレード
- Debian 12 での NVIDIA GPU/CUDA アクセラレーション サポートを備えた PyTorch のインストール
- PyTorch Python 3 仮想環境のアクティブ化
- PyTorch にアクセスし、NVIDIA GPU/CUDA アクセラレーションが利用可能かどうかを確認する
- 結論
Debian 12 への NVIDIA GPU ドライバーのインストール
PyTorch NVIDIA GPU/CUDA アクセラレーションが機能するには、次のことを行う必要があります。 Debian 12 に NVIDIA GPU ドライバーをインストールする 。 Debian 12 システムに NVIDIA GPU ドライバーをインストールする際にサポートが必要な場合は、 この記事を読む 。
Debian 12 への NVIDIA CUDA のインストール
PyTorch NVIDIA GPU/CUDA アクセラレーションを Debian 12 で動作させるには、次のことを行う必要があります。 Debian 12 に NVIDIA CUDA をインストールする 。 Debian 12 システムに NVIDIA CUDA をインストールする際にサポートが必要な場合は、 この記事を読む 。
Debian 12 への Python 3 PIP および Python 3 仮想環境 (venv) のインストール
Debian 12 に PyTorch をインストールするには、Python 3 PIP と Python 仮想環境 (venv) がインストールされている必要があります。
まず、次のコマンドを使用して APT パッケージ リポジトリ キャッシュを更新します。
$ 須藤 適切なアップデート 
Python 3 PIP および Python 3 仮想環境 (venv) をインストールするには、次のコマンドを実行します。
$ 須藤 適切な インストール python3-pip python3-venv python3-devインストールを確認するには、「Y」を押してから、 <入力> 。
Python 3 PIP と Python 3 venv がインストールされています。完了するまでにしばらく時間がかかります。
この時点で、Python 3 PIP と Python 3 venv がインストールされている必要があります。
PyTorch 用の Python 3 仮想環境の作成
Debian 12 に Python ライブラリをインストールする標準的な方法では、システムの Python パッケージ/ライブラリに干渉しないように、Python 仮想環境にそれらをインストールします。
PyTorch 用の新しい Python 3 仮想環境を「/opt/pytorch」ディレクトリに作成するには、次のコマンドを実行します。
$ 須藤 Python3 -m ベンヴ / 選択する / パイトーチPython 3 PyTorch 仮想環境での Python 3 PIP の最新バージョンへのアップグレード
Python 3「/opt/pytorch」仮想環境で Python 3 PIP を最新バージョンにアップグレードするには、次のコマンドを実行します。
$ 須藤 / 選択する / パイトーチ / 置き場 / pip3 インストール --アップグレード ピップ 
Debian 12 での NVIDIA GPU/CUDA アクセラレーション サポートを備えた PyTorch のインストール
PyTorch NVIDIA GPU/CUDA アクセラレーションが機能するには、Debian 12 システムにインストールした NVIDIA CUDA ドライバーのバージョンをサポートする正しいバージョンの PyTorch をインストールする必要があります。この記事の執筆時点では、PyTorch は NVIDIA CUDA ドライバー バージョン 11.8 および 12.1 をサポートしています。 PyTorch がサポートする NVIDIA CUDA ドライバーのバージョンに関する最新情報については、 PyTorch の公式 Web サイトを確認してください 。
Debian 12 システムにインストールした NVIDIA CUDA ドライバーのバージョンを確認するには、次のコマンドを実行します。ご覧のとおり、Debian 12 システムには NVIDIA CUDA バージョン 11.8 がインストールされています。
$ nvcc - バージョン 
NVIDIA CUDA 11.8 サポートを備えた PyTorch を PyTorch Python 3 仮想環境にインストールするには、次のコマンドを実行します。
$ 須藤 / 選択する / パイトーチ / 置き場 / pip3 インストール トーチ トーチビジョン トーチオーディオ --index-url https: // ダウンロード.pytorch.org / うーん / 118とNVIDIA CUDA 12.1 サポートを備えた PyTorch を PyTorch Python 3 仮想環境にインストールするには、次のコマンドを実行します。
$ 須藤 / 選択する / パイトーチ / 置き場 / pip3 インストール トーチ トーチビジョン トーチオーディオPyTorch は、PyTorch Python 3 仮想環境にインストールされます。完了するまでにしばらく時間がかかります。
この時点で、PyTorch は PyTorch Python 3 仮想環境にインストールされている必要があります。
PyTorch Python 3 仮想環境のアクティブ化
PyTorch Python「/opt/pytorch」仮想環境をアクティブにするには、次のコマンドを実行します。
$ 。 / 選択する / パイトーチ / 置き場 / 活性化PyTorch Python 3 仮想環境をアクティブ化する必要があります。
PyTorch にアクセスし、NVIDIA GPU/CUDA アクセラレーションが利用可能かどうかを確認する
Python 3 対話型シェルを開くには、次のコマンドを実行します。
$ Python3Python 3 対話型シェルが開かれている必要があります。
まず、次のコード行を使用して PyTorch をインポートします。
$ 輸入トーチ 
インストールした PyTorch のバージョンを確認するには、次のコード行を実行します。ご覧のとおり、NVIDIA CUDA 11.8 アクセラレーション サポート (cu118) を備えた PyTorch 2.1.0 を実行しています。
$ トーチ.__バージョン__ 
PyTorch が NVIDIA CUDA アクセラレーションのために NVIDIA GPU を使用できるかどうかを確認するには、次のコード行を実行することもできます。 NVIDIA CUDA サポートが利用可能な場合は、「True」と表示されます。
$ torch.cuda.is_available ( )コンピューターに複数の GPU がインストールされている場合は、次のコード行で PyTorch が使用できる GPU の数を確認できます。ご覧のとおり、Debian 12 システムには NVIDIA GPU (RTX 4070) がインストールされています。
$ トーチ.cuda.device_count ( )Python 対話型シェルを終了するには、次のコード行を実行します。
$ やめる ( )結論
この記事では、Python 3 PIP および Python 3 仮想環境 (venv) を Debian 12 にインストールする方法を説明しました。また、Debian 12 で PyTorch 用の Python 3 仮想環境を作成する方法と、NVIDIA CUDA を使用して PyTorch をインストールする方法も説明しました。 Debian 12 でも 11.8 および 12.1 アクセラレーションをサポートします。最後に、PyTorch Python 仮想環境をアクティブ化し、Debian 12 で PyTorch にアクセスする方法を説明しました。