Skip to main content

PyTorchのインストール方法

PyTorchのインストール方法について説明します。

最新バージョンをインストールする​

公式HPに、インストールしたい環境を入力することで、 インストールコマンドを知ることができます。基本的にはこれに従うことでインストールできます。

PyTorch getting start

過去のバージョンをインストールする​

公式のPrevious PyTorch Versionsを参考にインストールします。

PyTorch previous versions

Poetryを使ってインストールする​

pyproject.tomlを直接書き換える方法​

  1. pyproject.tomlを以下のように書き換えます。
    ただし、urlは、公式archiveやこちらなどを参考に取得します。

    [tool.poetry.dependencies]
    python = "^3.7"
    torch = {url = "https://download.pytorch.org/whl/cu116/torch-1.13.1%2Bcu116-cp37-cp37m-linux_x86_64.whl"}
    torchvision = {url = "https://download.pytorch.org/whl/cu116/torchvision-0.14.1%2Bcu116-cp37-cp37m-linux_x86_64.whl"}

  2. 以下のコマンドを実行してインストールします。

    poetry install

  3. 動作確認。

    以下のように、torchをインポートしGPUが認識しているか確認することができます。

    $ python
    Python 3.7.17 (default, Jun  6 2023, 20:10:10) 
    [GCC 9.4.0] on linux
    Type "help", "copyright", "credits" or "license" for more information.
    >>> import torch
    >>> torch.cuda.is_available()
    True

参考文献​