PyTorchのインストールガイド
PyTorchは、機械学習研究やアプリケーション開発に広く使用される強力なオープンソースの機械学習ライブラリです。ここでは、Pythonが既にインストールされている前提で、PyTorchをインストールする手順を解説します。
前提条件
- Pythonがインストール済みであること。未インストールの場合は、Python公式サイトからダウンロードしてインストールしてください。
PyTorchのインストール手順
-
PyTorchの公式サイトにアクセス
PyTorchのインストールは、公式サイトから行います。 -
環境の選択
- Stableバージョンを選択します。
- OS(例: Windows, macOS, Linux)を選びます。
-
Languageは
Python
を選択します。 -
CUDAのサポートが必要な場合(NVIDIA GPUを持っている場合)は、適切なCUDAバージョンを選択します。GPUがない場合は
CPU
を選択します。
-
コマンドのコピーと実行
サイトに表示されるコマンドをコピーし、Pythonがインストールされているコンピュータのコマンドラインまたはターミナルに貼り付けて実行します。
例:WindowsでCPUのみの場合のインストールコマンド
pip install torch torchvision torchaudio
例:WindowsでCUDA 12.4を利用する場合のインストールコマンド
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
インストール後の確認
インストールが完了したら、Python環境で以下のコマンドを実行してPytorchが正しくインストールされているかを確認します。
import torch
print(torch.__version__)
print("CUDA Available:", torch.cuda.is_available())
- CPUのバージョンCUDA Available出力: False
- GPUのバージョンCUDA Available出力: True
上記で出力が確認できたらPytorchのインストールは完了です。