Python

【Python】Windowsにpytorchをpipでインストールする

投稿日:2021年1月5日 更新日:

概要

CUDAなしの普通のWindowsマシンにpytorchをインストールする手順について試します。Anacondaではなく素のPython環境に入れます。

環境

以下で作ったpython3.8.7の仮想環境(venv)を使用します。

Step1 : Pytorchのpipコマンドの入手

pytorchは公式サイトのGetStartedのページ(https://pytorch.org/get-started/locally/)からpipコマンドを調べることができます。というのも、単にpip install ・・・をすればよいわけではなく、インストールする環境によって適切なコマンドがあるためです。今回は以下のような環境を選択の上で推奨pipコマンドを入手しました。

Pytorch Build : Stable(1.7.1)

Your OS : Windows

Package : pip

Language : Python

CUDA : None

Run this Commandが推奨のpipコマンドになりますのでこれをコピペします。

Step2 : pipインストール

コマンドラインからvenvの環境をActivateし、先ほどコピーしたコマンドをペースとして実行します。

pip install torch==1.7.1+cpu torchvision==0.8.2+cpu torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

Step3 : importして確認

Jupyterを立ち上げてimport torchを実行して無事に読み込めればとりあえずはOKです。

もっと詳細に確認したい場合は何かテスト用のコードなどを作って試すなど。

おまけ:アンインストールするとき

以下でアンインストールします。

pip uninstall torch

-Python

執筆者:


  1. kkano より:

    お世話になります。
    上記インストール画面を開くと、CUDA 11.0 が見受けられません。
    それを実用しているので、11.0 にしたいのですが、そうすればよいでしょうか?

    • SEBONE より:

      kkano様
      同環境で検証できていないので想像で恐縮ですがCUDA11.1用のpipコマンドでCUDA11.0でもビルドできないでしょうか?
      もしくは以下のようにpipコマンドを11.0用に手動で書き換えて実行するなどいかがですか。
      pip install torch==1.8.0+cu110 torchvision==0.9.0+cu110 torchaudio===0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

  2. kkano より:

    SEBONE様
    ありがとうございます。
    記載のhttpsに進むと、バージョンの異なる対応ファイルがいっぱい出てきますね。
    そこに合いそうなファイルがありました。
    これって、その該当するファイルをダブルクリックすると拡張子.whlファイルが
    ダウンロードできましたが、それをクリックするとインストールできるのでしょうか?
    実施したことがないのでご存じでしたら教えて下さい。

    • SEBONE より:

      kkano様
      目的のwhlファイルをダウンロードした後はコマンドラインからでローカルwhlのインストールができるはず。お試しください!
      pip install –no-index <ダウンロードしたwhlファイルのパス>

 Rapsberry Pi 4B にPyTorchとPyTorch Visionをインストール | いちりのテクの部屋 へ返信する コメントをキャンセル

メールアドレスが公開されることはありません。 * が付いている欄は必須項目です

関連記事

【Python】標準偏差と分散の計算

Contents1 目的2 標準偏差3 Pythonによる標準偏差の計算4 分散 目的 以前の記事で四分位偏差によりデータの分散度の計算を行いましたが、四分位偏差ではデータの代表値である四分位数のみを …

KerasからMNISTを取得して白黒表示するまで

Contents1 概要2 環境3 Step1 : MNISTの取得4 Step2 : MNISTを画像で表示5 Step3 : グレースケールで表示する6 参考)コード 概要 ディープラーニングの学 …

CentOSからAnacondaアンインストール

Contents1 目的2 Step1 : .pyenvの削除3 Step2 : bash_profileのpathの削除4 Step3 : .condaの削除5 Step4 : 確認 目的 以下の記 …

【Python】pandasでhtmlのtableをスクレイピング

Contents1 概要2 tableを取得するコード 概要 以下でBeautifulSoupを使ってwebページのhtmlからtableをスクレイピングしたときに少しはまったときの対処法について書き …

【Python】Spyderをインストールして構築済みのPython環境と接続する(Anaconda不使用)

Contents1 概要2 環境3 Step1 : Spyderのインストール4 Step2 : 立ち上げ&インタープリターの設定 概要 PythonのIDEとしてSpyderというものがあります。E …

言語切り替え

カテゴリー