前に逃げる

理系の大学院生。ディープラーニング初心者。

Windows10にpytorchをオフラインでインストール

f:id:dl-kento:20171227162318j:plain

pytorchとは

pytorchはchainerからフォークされたdefine-by-runのディープラーニングフレームワークで、主に研究開発用途で使われているみたいです。chainerからの派生だけあって非常に使い易いフレームワークです。

最近自分の周りでもpytorchを使う人が増えてきていて、勢いを増しているフレームワークです。

研究室にGeoforce GTX1080を積んだゲーミングPC(Arien Ware)があるのでそこにpytorchをインストールしましたが、一筋縄ではいかなかったので記録を残します。

今回は下のコマンドでAnacondaに仮想環境を作りその中で作業を行っています。

conda create -n pytorch python=3.6 numpy pyyaml mkl

結論から言うとcondaコマンドを使いオフラインでインストールするとできました。結果だけ見たい人は画面の下までスクロールしてください。

作業環境

  • OS: Windows10
  • GPU: GTX1080
  • Aconda: 4.4.0
  • python: 3.6.2

conda installでインストールできない!?

元々pytorchはLinuxmacOSしかサポートされておらず、公式からのインストールは出来ないのでAnaconda Cloudからのインストールになります。

conda install -c peterjc123 pytorch

このコマンドでインストールできるはずが、エラーが出て失敗した。。。

UnsatisfiableError: The following specifications were found to be in conflict:
  - pytorch
Use "conda info " to see the dependencies for each package.

pytorch元から入ってないしconflictする意味が分からん。ここから泥沼に嵌っていきました。以下失敗例。

  • Virtual Box で Ubuntu の環境を作ろうとしたが、失敗。Virtual Box からはGPUが認識できないみたいです。
  • wheelファイルをダウンロードしてきてpipでインストールしてみたが、import時にエラーが出た。

conflict を避けるため、offline でインストール

pytorch の GitHub issue を見ていたら「オフラインでpytorchをインストールしろ」との記述を発見。オフラインでインストールってどうやってやるんだとウェブを漁っていると中国語のページにたどり着きました。そこに書いてあった方法がこちら。

まず、Windows用のpytorchのソースコードをダウンロードし、そのディレクトリで以下のコマンドを実行します。

conda install --offline pytorch????.tar.bz2

これでいけました。ソースコードの最新版はAnacondaCloudに、レガシー版はこちらのページ(作者のGitHubページ)からリンクで飛べます。

まとめ

今回実感したのは中国語のページが非常にわかりやすかったということです。英語よりも、日本語よりもわかりやすかった。ディープラーニング人口が多いだけありますね。これからはわからないことは中国語のページも参考にしていこうと思います。

参考リンク

日本語ページ

qiita.com

abrakatabura.hatenablog.com

中国語ページ

PyTorch在64位Windows下的Conda包

www.ptorch.com