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

2020年10月22日木曜日

Anaconda Python PyTorch

t f B! P L

PyTorch

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

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

前提

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

【関連記事】
Anacondaのインストール (Mac OS編)

スポンサーリンク

PyTorchのインストール

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

仮想環境の作成

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

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

作成した仮想環境の三角形のボタンをクリックして、[Open Terminal]を選択します。
open terminalで仮想環境をターミナルで起動

作成したPythonの仮想環境でターミナルでが開きます。
作成した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]をクリックして、インストールします。

AnacondaでJupyter Notebookをインストール

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

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

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

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

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

さいごに

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

スポンサーリンク
スポンサーリンク

このブログを検索

Profile

自分の写真
Webアプリエンジニア。 日々新しい技術を追い求めてブログでアウトプットしています。
プロフィール画像は、猫村ゆゆこ様に書いてもらいました。

仕事募集もしていたり、していなかったり。

QooQ