DeepLearningライブラリPyTorchのインストール方法

DeepLearningライブラリの種類と選び方

PythonでサポートされているDeepLearningライブラリとしては……

  • PyTorch
  • TensorFlow
  • Chainer
  • Keras
  • Caffe

などが挙げられます。

いずれも優れたライブラリですが、環境設定・インストールの敷居が高いところがあります。

これらの中で、現在、最もお手軽にインストールができるライブラリはPyTorchです。

それでは、5分でPyTorchのインストールを済ませてしまいましょう。

PyTorchのインストール方法

  • インストール先のサーバの設定
    • Ubuntu 16.04 LTS
    • CUDA-8.0

Anacondaのインストール

はじめに、DeepLearning用のライブラリを集めたディストリビューションであるAnacondaというソフトウェアをインストールする必要があります。

Anacondaのダウンロードページにアクセスし、Anacondaのリンクを取得します。

wgetコマンドでリンク先をダウンロードし、その後実行権限を与えてインストール。

PyTorchをインストール

PyTorchの公式サイトからget-startedのページにアクセスし、以下の要件でPyTorchのインストール用コマンドを生成します。

生成したコマンドを実行して、PyTorchのインストールが完了します。

たったこれだけでPyTorchによるDeepLearningが実行できるようになりました!

chainerやTensorFlowに比べ、あっという間に、かつ安定して環境構築ができるところがPyTorchの大きなメリットです。

PyTorchのテスト

それでは、インストールしたPyTorchがGPUモードで動くことの確認をしてみます。

ipythonのインタープリタから、以下のコードを実行してみましょう。

無事にGPUでベクトルの計算ができたことが確認できました。

これで好きなだけPyTorchを使ってDeepLearningを実行させることができます。

参考

  • Anaconda is the most popular Python data science platform with 6 million users. Anaconda Enterprise enables data science teams to collaborate, share and deploy ...
  • An open source deep learning platform that provides a seamless path from research prototyping to production deployment.