Anacondaに「PyTorch」をインストールする

PyTorch

機械学習の人気ライブラリには、「TensorFlow」「Scikit-learn」などがありますが、2016年にリリースされ、「PyTorch」が人気上昇中で注目を集めています。
この記事では、Anacondaで「PyTorch」をインストールする方法を解説します。

【PyTorch公式】
▶https://pytorch.org/

前提

Mac OSで前提です。といっても、Windowsでもインストール方法は同じだと思います。
Anaconda本体はインストールされている前提です。もしインストールされていなければ、以下の記事を参考にインストールしてください。

【関連記事】
▶Anacondaのインストール （Mac OS編）

スポンサーリンク

PyTorchのインストール

では、Anacondaで「PyTorch」をインストールする方法を見ていきましょう。

仮想環境の作成

デフォルトの仮想環境にインストールしてもいいですが、せっかくなので、PyTorch用の仮想環境を作りましょう。
仮想環境は、左メニューの[Environment]から作成します。

作成した仮想環境でターミナルを起動

作成した仮想環境の三角形のボタンをクリックして、[Open Terminal]を選択します。

作成したPythonの仮想環境でターミナルでが開きます。

PyTorchをインストール

ターミナル上で、以下のコマンドを入力してPyTorchをインストールします。

conda install pytorch torchvision -c pytorch

以下のように表示されれば、インストールOKです。

~ 中略 ~
mkl_fft-1.1.0        | 135 KB    | ##################################### | 100% 
Preparing transaction: done
Verifying transaction: done
Executing transaction: done

Matplotlibのインストール

Matplotlibは、Pythonのグラフ描画ライブラリです。
Matplotlibを使うと2次元、3次元等のテータを簡単にブラフにプロットすることができ、Pythonでデータ分析をする時には、必須のツールといえます。

conda install matplotlib

以下のように表示されれば、インストールOKです。

~ 中略 ~
matplotlib-base-3.2. | 5.4 MB    | ##################################### | 100% 
Preparing transaction: done
Verifying transaction: done
Executing transaction: done

スポンサーリンク

動作確認

PyTorchのインストールをしたら、正常に動作するか確認しましょう。定番の「Jupyter Notebook」を使って動作を確認します。

Jupyter Notebookのインストール

左メニューの[Home]から、「Jupyter Notebook」の[Install]をクリックして、インストールします。

インストールが完了したら、左メニューの[Environment]を選択し、PyTorch用の仮想環境で[Open WIth Jupyter Notebook]を選択します。

「Jupyter Notebook」が起動すると、ブラウザが立ち上がります。右上の「New」ボタンをクリックして、新規に作成します。

PyTorchが正常にインストールされているか確認するため、コード入力エリアに、乱数を取得する次のコードを入力します。

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

入力したら、Shift + Enterキーを押して、コードを実行します。
次のような乱数が結果に表示されれば、インストール成功です。

さいごに

AnacondaでPyTorchをインストールする方法を紹介してきました。
「Jupyter Notebook」を使えば、コードの実行結果を確認しながら学習できるため、便利です。