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

概要

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

環境

• python3.8.7(venvによる仮想環境)

以下で構築した環境になります。

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

pytorchは公式サイトのGetStartedのページ(https://pytorch.org/get-started/locally/)からpipコマンドを調べることができます。というのも、単にpip install torchをするよりも、環境によって適切なコマンドを使用してインストールするのがベストであるためです。

今回は以下のような環境を選択の上で推奨pipコマンドを入手しました。

Pytorch Build : Stable(1.7.1)

Your OS : Windows

Package : pip

Language : Python

CUDA : None

今回のテスト環境ではCUDAを使用していませんが、CUDAでGPUを使用したい場合は適切なCUDAのバージョンを選択するようにしてください。

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