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

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

目次

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

サイト内の関連Webページ

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


前準備

Anaconda のインストール

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

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

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

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


PyTorch のインストール

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

  2. オプション)Anaconda をインストールしていて、Anaconda の Python 環境を有効しているときの追加の操作

    Anaconda の Python 環境に OpenCV をインストールしたいときは、 Anaconda の Python 環境を有効する

    「activate ai 」は Python 環境(名前はai)を有効にするためのコマンド(各自の環境の名前にあわせること

    activate ai 
    

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

  3. Anaconda の conda パッケージの更新
    conda upgrade --all
    

  4. PyTorch, torchvision, CUDA, CuDNN のインストール

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

    conda install pytorch torchvision cuda100 cudnn -c pytorch
    

    (以下省略)

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

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

    conda config --remove channels pytorch
    

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

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

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

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

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