【PyTorch】インストール方法(Windows, Mac OSX編)

Pythonの深層学習ライブラリ「PyTorch」をWindowsやMac OSX上にインストールする方法についてまとめました。

【導入方法１】PyTorchのインストール（Windows編）

● コマンドプロンプトを管理者として実行します。

● 以下のコマンドでインストールされているPython のバージョンを確認します。

python --version

● PyTorch の公式ページ（https://pytorch.org/）にアクセスします。

● 次のように選択し、その下に表示されるランタイムコマンドをコピーします。

● コピーしたランタイムコマンドを、コマンドプロンプトで実行します。

● PyTorch の公式ページ（https://pytorch.org/）にアクセスします。

● 次のように選択し、その下に表示されるランタイムコマンドをコピーします。

–	–
Your OS	Mac
Package	pip
Language	Python
CUDA	使用している CUDA のバージョン（使っていなければNone）

● コピーしたランタイムコマンドを、ターミナルで実行します。

※エラーが出る場合は、以下のコマンドでpipやsetuptoolsをアップデートしてみてください

pip3 install --upgrade pip
pip3 install setuptools --upgrade

● 以下のスクリプトを実行しエラーが出なければインストール成功です。

import torch
print(torch.__version__)

pipで深層学習ライブラリ「PyTorch」をインストールする手順は下記の通りです。

① NumPyのインストールをしていない場合、先に入れておきます。

–	NumPyのインストール方法はこちら
参考	■NumPyのインストール (Windows、Linux、Ubuntu)

③ whlファイルをクリックしてダウンロードします。
※お使いのPythonバージョン、利用したいOpenCVのバージョンによって選択します。

ケース	ダウンロードするwhlファイル名
Python3.6(64bit)にPyTorchをインストールしたい場合	torch‑0.4.0‑cp36‑cp36m‑win_amd64.whl
Python3.7(64bit)にPyTorchをインストールしたい場合	torch‑0.4.0‑cp37‑cp37m‑win_amd64.whl

④ ダウンロードが終わったらコマンドプロンプトを起動します。

⑤ コマンドプロンプトでダウンロードしたファイルがあるフォルダ（ディレクトリ）まで移動します。

⑥ 以下のコマンドを打ち込みます。

pip install torch‑0.4.0‑cp36‑cp36m‑win_amd64.whl

※ファイル名はダウンロードした物に応じて適宜変更してください

⑦ 以上でインストール完了です。
以下のスクリプトを実行しエラーが出なければインストール成功です。

import torch
print(torch.__version__)

pipで深層学習ライブラリ「PyTorch」をWinPythonでインストールする手順は下記の通りです。

② whlファイルをクリックしてダウンロードします。
※お使いのPythonバージョン、利用したいOpenCVのバージョンによって選択します。

ケース	ダウンロードするwhlファイル名
WinPython3.6(64bit)にPyTorchをインストールしたい場合	torch‑0.4.0‑cp36‑cp36m‑win_amd64.whl
WinPython3.7(64bit)にPyTorchをインストールしたい場合	torch‑0.4.0‑cp37‑cp37m‑win_amd64.whl

③「WinPython Control Panel.exe」を起動します。

④ メニューから[Packages]→[Add packages]をクリックします。

⑤ ダウンロードしたwhlファイルを選びます。

⑥ [Install packages]ボタンを押して、画面から選択したファイル名が消えたらインストール完了です。

⑦ 以下のスクリプトを実行しエラーが出なければインストール成功です。

import torch
print(torch.__version__)

–	参考
詳細	■【WinPython】PyTorchのインストール

–	関連記事
1	■【Python/PyTorch入門】ディープラーニングの使い方
2	■【機械学習入門】アルゴリズム＆プログラミング
3	■Python入門基本文法