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

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

【導入方法1】PyTorchのインストール(Windows編)

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

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

python --version

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

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

Your OS Windows
Package pip
Language Python
CUDA 使用している CUDA のバージョン

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

【導入方法2】PyTorchのインストール(Mac OSX編)

● 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__)

【導入例2】PyTorchをpip + whlファイルでインストール

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

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

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

② 以下のURL先にアクセスします。
http://www.lfd.uci.edu/%7Egohlke/pythonlibs/#pytorch

③ 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__)

【導入例3】PyTorchをpip+whlファイルでインストール(WinPython編)

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

① 以下のURL先にアクセスします。
http://www.lfd.uci.edu/%7Egohlke/pythonlibs/#pytorch

② 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入門 基本文法

コメント