機械学習
DeepLearning
PyTorch

PyTorch 最初の1歩(開発環境構築)

概要

プログラム経験のない友人に深層学習を教えるための資料を作成しています。
せっかく作成したので公開してみることにしました。

  • 今回の目標
    • Windows環境でPyTorchを使用します。
      • Anacondaインストール
      • PyTorchインストール
      • サンプルプログラムmnistの動作確認

Anaconda

  • Anacondaを使用することで、Pythonと多くのライブラリをまとめてインストールできます。
  • 2018/1/14時点のバージョンはAnaconda 5.0.1。
  • https://www.anaconda.com/download/ からダウンロードしてください。 anaconda1.png

Pytorch

  • Anaconda Promptで以下のコマンドを入力するとインストールされます。
conda install -c peterjc123 pytorch
  • インストール中にUpdateしてもいいかどうかの許可を求められるので、'y'を入力してください。
    PyTorch2.png

  • Anaconda Promptで以下のコマンドを入力して、torchvisionもインストールします。

pip install torchvision

PyTorch3.png

動作確認

  • https://github.com/pytorch/examples の"Clone or download"から"Download ZIP"を選択してダウンロードし、zipファイルを解凍します。
  • 解凍したフォルダ一式はAnacondaのHome階層に置きます。
  • Anaconda Promptで/example-master/mnistへ移動します。(コマンドcd example-master/mnist)
  • Anaconda Promptで以下のコマンドを入力します。
python main.py
  • 学習が開始されます。Train Epochは10回目で終了します。

PyTorch4.png

以上が正常に動作すれば、環境構築は成功です。
次回は、PyTorchの基本的な文法を確認しながら、パーセプトロンの実装を試みたいと思います。

参考サイト