トップページ -> 最新情報技術の実習と研究ツール -> Windows でソフトウエアのインストールと設定 -> Windows で PyTorch をインストールする
[サイトマップへ]

Windows で PyTorch をインストールする

目次

PyTorch など、インストールするソフトウエアの利用条件などは、利用者が確認すること。

サイト内の関連Webページ

謝辞:このWebページで紹介する PyTorch ソフトウエア及びその他のソフトウエアの作者に感謝します


前準備

Anaconda のインストール

前準備として,Python 開発環境のAnaconda のインストールが終わっていること.

以下,Windows に Anaconda をインストール済みであるものとして説明を続けます.

(オプション)NVIDIA グラフィックスカード・ドライバのインストール

CUDA と連携させたいときは、前もって、 NVIDIA グラフィックスカード・ドライバ, NVIDIA CUDA ツールキットをインストールしておく.


PyTorch のインストール

NVIDIA グラフィックスカード・ドライバ, NVIDIA CUDA ツールキットのインストール

CUDA と連携させたいときは、前もって、 NVIDIA グラフィックスカード・ドライバ, NVIDIA CUDA ツールキットをインストールしておく.

PyTorch のインストール

  1. Windows のコマンドプロンプトを実行する.

  2. PyTorch, torchvision のインストール

    ※ conda がないときは、conda のインストールを行う

    conda install pytorch torchvision -c pytorch
    

    (以下省略)

  3. conda のチャネル削除操作

    エラーメッセージは無視して良い

    conda config --remove channels pytorch
    

  4. Python でPyTorch のバージョン確認

    python -c "import torch; print( torch.__version__ )"
    

  5. 動作確認 https://pytorch.org/get-started/locally/ に記載の Python プログラムを動かしてみる

    from __future__ import print_function
    import torch
    x = torch.rand(5, 3)
    print(x)
    

    次のような結果が表示されることを確認。乱数を使っているので、値は、実行のたびに変化する