UbuntuにPyTorchをインストールする手順【CUDA/CPU両対応】

GPU

UbuntuにPyTorchをインストールしたいけれど、どのコマンドを打てばいいのかよく分からない——そんな方に向けた記事です。

結論から言うと、Ubuntu 24.04 LTS では pip3 install torch を直接実行するとエラーになります(PEP 668 という制約のため)。正しい手順は 仮想環境(venv)を作ってから、その中で pip install する方法です。

本記事では、実際に Docker 上の Ubuntu 24.04 で pip install torch torchvision torchaudio を最後まで走らせて取得した実測ログをもとに、CPU版・CUDA版それぞれのインストール手順を解説します。想像で書いた数字ではなく、本物の実行結果だけを載せています。

この記事のポイント

  • Ubuntu 24.04 では pip3 install torch の直接実行は NG(externally-managed-environment エラー)→ python3 -m venv で仮想環境が必須
  • CPU版は --index-url https://download.pytorch.org/whl/cpu を付ける。torch 本体の whl は実測で約 192MB、依存込みで venv は約 978MB に膨らむ
  • 2026年6月時点の最新は torch 2.12.0 / torchvision 0.27.0 / torchaudio 2.11.0(実インストールで確認)
  • CUDA版は nvidia-smi の CUDA バージョンに合わせて cu126 等を選ぶ。最新 torch を GPU で使うなら cu124 ではなく cu126
  • 動作確認は python -c "import torch; print(torch.__version__)" で一発

目次

  1. 前提環境と確認
  2. Python・pipのインストール
  3. 仮想環境(venv)の作成
  4. PyTorch CPU版のインストール
  5. PyTorch CUDA版のインストール
  6. インストール確認・動作テスト
  7. よくあるエラーと解決策
  8. まとめ

前提環境と確認

本記事の検証は、Docker 公式イメージ(ubuntu:24.04 / ubuntu:22.04)を使って実行しました。VPSやWSLでも同じ手順がそのまま使えます。まずは Ubuntu のバージョン別に、標準で入る Python と pip を実測で比較します。

項目 Ubuntu 24.04 LTS Ubuntu 22.04 LTS
Pythonバージョン(実測) 3.12.3 3.10.12
pip バージョン(実測) 24.0 22.0.2
python3-pip パッケージ版(実測) 24.0+dfsg-1ubuntu1.3 22.0.2+dfsg-1ubuntu0.7
サポート期限 2029年4月(推奨) 2027年4月
externally-managed制約 あり(PEP 668) なし
Ubuntu 22.04 vs 24.04 の Python/pip バージョン実測比較
Ubuntu 22.04 vs 24.04 の Python/pip バージョン実測比較

上の数字は、docker run --rm ubuntu:24.04ubuntu:22.04 をそれぞれ起動して python3 --version / pip3 --version / apt-cache policy python3-pip を実行した実測値です。Ubuntu 24.04 では Python 3.12 が標準になり、ここで紹介する PEP 668 の制約が入りました。

注意

本記事のコマンドは Ubuntu 24.04 LTS を主軸に検証しています。Ubuntu 22.04 では Python が 3.10 で PEP 668 の制約もないため、一部の手順が不要になります(記事内で都度補足します)。

Python・pipのインストール

Ubuntuの公式イメージや最小インストール環境では、最初から python3-pip が入っていないことがあります。まず確認して、なければインストールします。

手順1:Python・pipの確認

ubuntu@linuxlab: ~
$ python3 –version Python 3.12.3 $ pip3 –version pip 24.0 from /usr/lib/python3/dist-packages/pip (python 3.12)

もし pip3: command not found と出る場合は、次の手順でインストールします。

手順2:python3-pip・python3-venv のインストール

ubuntu@linuxlab: ~
$ sudo apt update Reading package lists… Done $ sudo apt install python3-pip python3-venv Get:3 … libpython3.12-minimal 3.12.3-1ubuntu0.13 [833 kB] Get:5 … python3.12-minimal 3.12.3-1ubuntu0.13 [2265 kB] Get:17 … python3 3.12.3-0ubuntu2.1 [23.0 kB] Setting up python3-pip (24.0+dfsg-1ubuntu1.3) … Setting up python3-venv (3.12.3-0ubuntu2.1) …
python3-pip / python3-venv インストール実ログ(Ubuntu 24.04 実測)
python3-pip / python3-venv インストール実ログ(Ubuntu 24.04 実測)

インストールが完了すると、Ubuntu 24.04 では Python 3.12.3 / pip 24.0、Ubuntu 22.04 では Python 3.10.12 / pip 22.0.2 が使えるようになります。

著者アイコン
著者アイコン

python3-venv を忘れがちですが、後で仮想環境を作るために必要です。python3-pip と一緒にインストールしておきましょう。

仮想環境(venv)の作成

ここが Ubuntu 24.04 で最もつまずきやすいポイントです。pip3 install を直接実行すると、次のようにエラーになります。

ubuntu@linuxlab: ~
$ pip3 install torch error: externally-managed-environment × This environment is externally managed ╰─> To install Python packages system-wide, try apt install python3-xyz, where xyz is the package you are trying to install. If you wish to install a non-Debian-packaged Python package, create a virtual environment using python3 -m venv path/to/venv. Then use path/to/venv/bin/python and path/to/venv/bin/pip. Make sure you have python3-full installed. note: If you believe this is a mistake, please contact your Python installation or OS distribution provider. … by passing –break-system-packages. hint: See PEP 668 for the detailed specification.

これは Ubuntu 24.04 が導入した「PEP 668」という仕組みによるものです。OSが管理する Python 環境を pip が壊さないように、システムへの直接インストールをブロックしています。上のメッセージにも書かれているとおり、解決策は python3 -m venv で仮想環境を作ることです。

手順3:仮想環境を作成して有効化する

ubuntu@linuxlab: ~
$ python3 -m venv ~/pytorch-env $ source ~/pytorch-env/bin/activate (pytorch-env) $ (pytorch-env) $ pip –version pip 24.0 from /home/user/pytorch-env/lib/python3.12/site-packages/pip (python 3.12)

プロンプトが (pytorch-env) $ になれば、仮想環境が有効化されています。以降の pip コマンドはすべてこの環境の中だけで動き、システムの Python を汚しません。

仮想環境を抜けるには

deactivate を実行するとシステムの Python 環境に戻ります。次回 PyTorch を使うときは、再度 source ~/pytorch-env/bin/activate で有効化してください。

PyTorch CPU版のインストール

GPUなし・学習目的・小規模なデータ処理なら、CPU版で十分です。CUDA版より手軽に試せます。インストールには PyTorch 公式の配布インデックスを --index-url で指定します。公式の「Start Locally」ページでも、この CPU 用インデックスURLが案内されています。

PyTorch 公式「Start Locally」ページ(Playwright 実撮影 2026-06-15・最新安定版は 2.12.0)
PyTorch 公式「Start Locally」ページ(Playwright 実撮影 2026-06-15・最新安定版は 2.12.0)

公式ページのとおり、2026年6月時点の最新安定版は PyTorch 2.12.0 です。「Latest Stable PyTorch requires Python 3.10 or later」とも明記されており、Ubuntu 24.04(Python 3.12)/ 22.04(Python 3.10)はどちらも対応範囲です。

手順4:torch / torchvision / torchaudio をインストール

ubuntu@linuxlab: ~ (pytorch-env)
(pytorch-env) $ pip install torch torchvision torchaudio \ –index-url https://download.pytorch.org/whl/cpu Downloading torch-2.12.0+cpu-cp312-cp312-manylinux_2_28_x86_64.whl (192.3 MB) Downloading torchvision-0.27.0+cpu-cp312-cp312-manylinux_2_28_x86_64.whl (1.8 MB) Downloading torchaudio-2.11.0+cpu-cp312-cp312-manylinux_2_28_x86_64.whl (341 kB) Downloading numpy-2.4.4-cp312-cp312-manylinux_2_28_x86_64.whl (16.6 MB) Downloading sympy-1.14.0-py3-none-any.whl (6.3 MB) # ほか networkx / pillow など Successfully installed torch-2.12.0+cpu torchvision-0.27.0+cpu torchaudio-2.11.0+cpu numpy-2.4.4 sympy-1.14.0 networkx-3.6.1 pillow-12.2.0 …
venv + PyTorch CPU版 実インストールログ(Ubuntu 24.04 / x86_64 実測)
venv + PyTorch CPU版 実インストールログ(Ubuntu 24.04 / x86_64 実測)

インストールに成功すると、torch 2.12.0+cpu / torchvision 0.27.0+cpu / torchaudio 2.11.0+cpu が入ります(2026-06-15、docker の x86_64 環境で実測)。バージョン末尾の +cpu が、CPU版である目印です。

ディスク容量に注意:torch本体だけで約192MB、venvは約1GB

実測したところ、torch 本体の whl は 192.3MB ありました。さらに numpy・sympy・networkx・pillow などの依存も一緒に入り、インストール後の venv は合計 978MB(torch ディレクトリだけで 750MB)になります。ストレージの小さいVPS(10〜25GBプランなど)では、空き容量に余裕を持たせておきましょう。

PyTorch CPU版 実インストール結果(バージョン・容量の実測)
PyTorch CPU版 実インストール結果(バージョン・容量の実測)

「公式インデックスに本当にこの whl があるの?」と気になる方は、https://download.pytorch.org/whl/cpu をブラウザで開くと配布ファイル一覧が確認できます。

PyTorch CPU版 whl 配布インデックス(Playwright 実撮影 2026-06-15)
PyTorch CPU版 whl 配布インデックス(Playwright 実撮影 2026-06-15)

PyTorch CUDA版のインストール

NVIDIA GPU を積んだ環境(GeForce RTX シリーズ、VPS の GPU プランなど)では CUDA 版を使うことで、学習・推論が大幅に高速化します。CPU版との違いは --index-url の末尾だけです。

CUDA版インストールの前提

CUDA版の PyTorch を動かすには、NVIDIA ドライバが必要です(CUDAライブラリ本体は torch の whl に同梱されるため、別途 CUDA Toolkit を入れる必要はありません)。まず nvidia-smi でドライバが対応する CUDA バージョンを確認してから、対応する index URL を選びます。

手順A:NVIDIA ドライバ・CUDAバージョンを確認する

ubuntu@linuxlab: ~ (GPU搭載マシンでの実行例)
$ nvidia-smi +—————————————————————————–+ | NVIDIA-SMI 560.35.03 Driver Version: 560.35.03 CUDA Version: 12.6 | +—————————————————————————–+ # 「CUDA Version」の数字に対応した –index-url を選ぶ

※上の nvidia-smi は GPU 搭載マシンでの出力例です(本記事の検証環境はGPUなしのため)。

nvidia-smi の「CUDA Version」が、そのドライバでサポートする最大の CUDA バージョンです。次に、PyTorch 公式インデックスを実際に調べて作った対応表を見てください。

CUDA別インデックスで入手できる最新 torch(公式インデックス実取得)
CUDA別インデックスで入手できる最新 torch(公式インデックス実取得)

これは download.pytorch.org/whl/<index>/torch/ を実際に取得し、入手できる torch の最新版を抽出した実測の対応表です(2026-06-15)。ここから分かる重要なポイントが2つあります。

index URL 選びの実測ポイント

  • 最新の torch 2.12.0 を GPU で使いたいなら cu126(CUDA 12.6)を選ぶ。cpucu126 だけが 2.12.0 まで追従している
  • cu124 は torch 2.6.0、cu121 は 2.5.1 で更新が止まっている。「いつもの cu124」を選ぶと古い torch しか入らない点に注意

手順B:CUDA 12.6 対応の PyTorch をインストールする

ubuntu@linuxlab: ~ (pytorch-env)
(pytorch-env) $ pip install torch torchvision torchaudio \ –index-url https://download.pytorch.org/whl/cu126 Looking in indexes: https://download.pytorch.org/whl/cu126 # CUDA版の whl は CUDA ライブラリを同梱するため、CPU版(192MB)より # さらに大きく、数GB規模のダウンロードになります

CUDA版はCPU版よりはるかに大きい

CPU版の torch 本体が約192MBだったのに対し、CUDA版は cuDNN や cuBLAS などの CUDA ライブラリを同梱するため、ダウンロード・ディスク使用量ともに数GB規模になります。VPS環境では通信量とストレージ残量に注意してください。インストール中しばらく時間がかかるのは正常です。

インストール確認・動作テスト

インストールが終わったら、本当に使える状態か確認します。次の3つを実行して、すべてエラーなく通れば成功です。

手順5:import torch で動作確認

ubuntu@linuxlab: ~ (pytorch-env)
(pytorch-env) $ python -c “import torch; print(torch.__version__)” 2.12.0+cpu (pytorch-env) $ python -c “import torch; print(torch.cuda.is_available())” False (pytorch-env) $ python -c “import torch; torch.manual_seed(0); print(torch.rand(2,3))” tensor([[0.4963, 0.7682, 0.0885], [0.1320, 0.3074, 0.6341]])
PyTorch 動作確認コマンドの実出力(torch.__version__ / CUDA判定 実測)
PyTorch 動作確認コマンドの実出力(torch.__version__ / CUDA判定 実測)

import torch でエラーが出なければ、正常にインストールされています。CPU版なので torch.cuda.is_available()False ですが、これは正常です(CUDA版+NVIDIAドライバの環境なら True になります)。torch.manual_seed(0) でシードを固定すると、誰の環境でも torch.rand は同じ値(0.4963, 0.7682, ...)を返すので、正しく動いているかの目安になります。

デバイスを自動判定するコード例

pytorch-env — test_device.py
$ cat test_device.py import torch device = ‘cuda’ if torch.cuda.is_available() else ‘cpu’ print(f’PyTorch バージョン: {torch.__version__}’) print(f’使用デバイス: {device}’) x = torch.rand(3, 3).to(device) print(f’テンソルのデバイス: {x.device}’) $ python test_device.py PyTorch バージョン: 2.12.0+cpu 使用デバイス: cpu テンソルのデバイス: cpu

このコードは CPU/CUDA を自動で切り替えるので、環境に関係なく同じスクリプトが動きます。学習スクリプトの冒頭に置いておくと便利です。CUDA版+GPUのある環境で実行すれば、使用デバイス: cuda と表示されます。

なお、参考までに本記事の検証を回した共有Dockerホストの sysbench CPU ベンチも載せておきます(PyTorch の CPU 推論性能はコア数や命令セットに左右されるため、実運用ではご自身のVPSで計測するのが確実です)。

検証環境のCPUベンチ(sysbench 実測・参考値)
検証環境のCPUベンチ(sysbench 実測・参考値)

よくあるエラーと解決策

①エラー:externally-managed-environment

ubuntu@linuxlab: ~
$ pip3 install torch error: externally-managed-environment

原因:Ubuntu 24.04 の pip は PEP 668 により、システムの Python 環境への直接インストールをブロックします。

解決策python3 -m venv ~/pytorch-env && source ~/pytorch-env/bin/activate で仮想環境を作ってから pip install を実行してください。どうしてもシステムに直接入れたい場合のみ --break-system-packages がありますが、OSのPythonを壊すリスクがあるため非推奨です。

②エラー:ModuleNotFoundError: No module named ‘torch’

ubuntu@linuxlab: ~
$ python3 -c “import torch” ModuleNotFoundError: No module named ‘torch’

原因:仮想環境を有効化せずに python3 を実行している可能性が高いです。

解決策source ~/pytorch-env/bin/activate で仮想環境を有効化してから実行してください。プロンプトに (pytorch-env) が表示されているか確認します。

③エラー:torch.cuda.is_available() が False になる

原因1:CPU版(torch-2.12.0+cpu)をインストールしている。バージョン末尾が +cpu だと CUDA は使えません。GPU版(cu126 等)を入れ直す必要があります。

原因2:NVIDIA ドライバが入っていない。nvidia-smi が通らない場合はドライバから入れ直してください。

確認コマンドpip list | grep torchtorch 2.12.0+cpu と出ていれば CPU版、torch 2.12.0+cu126 のように +cuXXX なら CUDA 版です。

④エラー:pip install 時に「Could not find a version that satisfies」

ubuntu@linuxlab: ~ (pytorch-env)
(pytorch-env) $ pip install torch==2.12.0 –index-url https://download.pytorch.org/whl/cu124 ERROR: Could not find a version that satisfies the requirement torch==2.12.0

原因:指定したインデックスに、そのバージョンの whl が存在しないためです。たとえば cu124 は torch 2.6.0 で更新が止まっているため、2.12.0 は取得できません。

解決策:手順Aの実測対応表を参考に、欲しい torch バージョンを配布しているインデックスを選んでください。最新の 2.12.0 が欲しいなら cpucu126 を使います。

まとめ

UbuntuへのPyTorchインストールは、バージョンによって手順が変わります。実測で分かった要点をまとめます。

  • Ubuntu 24.04 LTS(Python 3.12.3)では pip の直接実行が NG → python3 -m venv で仮想環境を作ってからインストール
  • CPU版コマンド:pip install torch torchvision torchaudio –index-url https://download.pytorch.org/whl/cpu
  • CPU版でも torch 本体は約192MB、依存込みで venv は約978MB。VPSの空き容量に注意
  • 2026年6月時点の最新は torch 2.12.0 / torchvision 0.27.0 / torchaudio 2.11.0(実インストールで確認)
  • CUDA版は nvidia-smi の CUDA バージョンを確認 → 最新 torch を狙うなら cu124 ではなく cu126 を選ぶ
  • 動作確認は python -c “import torch; print(torch.__version__)” で一発。+cpu か +cuXXX かでどちらの版か分かる

PyTorchを本格的にGPUで動かすなら、GPU搭載のVPSが初期費用ゼロで始められて手軽です。自前でGPUマシンを組むより、まずはクラウドのGPUプランで試すのがおすすめです。

VPS各社のGPUプラン・料金の比較は、以下の記事も参考にしてみてください。

GPU搭載VPS比較 ── 機械学習・AI推論向けクラウドの選び方

コメント

タイトルとURLをコピーしました