以前はWindows 10でディープラーニングするのに、AnacondaでPython環境を作ってTensorFlowを使ってVisual Studioでコーディングしてたけど、
最近はだいぶ状況が変わってきた。
Anacondaでnumpyが動かなくなったり、使いたいパッケージ(のバージョン)がcondaで入らなかったりで、そもそもAnacondaのメリットよりもデメリットで時間を取られることが増えた。(それはMinicondaでも同じ)
なので、Pythonは普通に公式のインストーラーでWindowsにインストールすることにした。
コーディング環境についても、VSCode(Visual Studio Code)が優秀だと気づいた。
VSCodeはIDEじゃなくてエディタなんだけど、Extensionをインストールして機能拡張はできる。オイラがIDEの方のVisual Studioを使っていた理由の8割ぐらいはIntelliSense(コード補完機能)だったので、同じくIntelliSenseが載っているなら動作も軽いVSCodeの方が快適なのでした。
そして、ディープラーニングのライブラリはTensorFlowからPyTorchへ乗り換えつつある。
理由はそんなにハッキリとしたものじゃないんだけど、最近は発表される論文と一緒に公開されるソースコードにPyTorch実装のものが増えている点と、対応する依存関係(CUDAのバージョンとか)のアップデートの早さとか。最近やっとバージョン1に達して、身近な人達がどんどんPyTorchへ移行しているというのもある。
ということで、今現在のオイラの環境は以下↓
- Windows10 Pro
- CUDA Toolkit 10.1
- cuDNN v7.5.0 (Feb 25, 2019), for CUDA 10.1
- Python 3.6.8
- PyTorch 1.0.1
- VSCode
一応、インストール手順もメモしておく。
PyTorch 1.0.1は現時点で最新のCUDA Toolkit 10.1とcuDNN v7.5に対応しているのであんまり気にしなくて良いけど、この手のライブラリのバージョンの依存関係は間違えるとエラー地獄にはまるのでインストール前にしっかり確認した方が良いです。(何度かハマった)
追記:PyTorchとtensorboardXを一緒に使いたい場合は、TensorBoard(というかTensorFlow)の動作環境にも合わせてやる必要があります。
https://www.tensorflow.org/install/gpu
CUDAとcuDNNのインストール
CUDAの公式ページからCUDA Toolkit 10.1のインストーラーをダウンロードしてインストール。
cuDNN v7.5.0 (Feb 25, 2019), for CUDA 10.1は、まずMembership登録してログインし、cuDNN一式のzipをダウンロードする。
そしてzipを展開して出てきたbin, include, libディレクトリを丸ごとCUDAのパスが通ったディレクトリに配置する。オイラはいつもCUDAデフォルトのインストールパスに一緒に突っ込んでいる↓
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
ところで、cuDNNってCUDA Deep Neural Network libraryの略なんですね。
Pythonのインストール
Pythonはバージョン3.6を使う。公式ページからPython3.6.8のWindows x86-64 executable installerをダウンロードしてきて実行するだけ。
インストールパスだけデフォルトから以下に変えておいた。
C:\dev\Python36
PyTorchのインストール
PyTorchは、公式ページのSTART LOCALLYで自分の環境を選ぶと適切なpipのコマンド表示されるので、それを叩けばインストールできる↓
ということで、以下のコマンドをコマンドプロンプトで実行するだけでPyTorchのインストール完了↓
pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.1-cp36-cp36m-win_amd64.whl
pip3 install torchvision
VSCodeのインストール
VSCodeは公式ページからWindows用のインストーラーをダウンロードしてインストール。
そして日本語の言語パックとPython用のExtensionもインストールする。
これでPyTorch環境は整ったので、あとは適当なサンプルコードがちゃんと実行できることを確認してみよう。
GitHubにPyTorch公式のサンプルコードがある↓
https://github.com/pytorch/examples
Basic MNIST Exampleとかが手軽で良いんじゃないでしょうか。
サンプルが実行できたらとりあえずOKだ。
追記:最近は、PyTorchに特化したIDEもある↓
PyTorchの習得
PyTorch公式サイトのサンプルで学ぶPyTorchや、
https://pytorch.org/tutorials/beginner/pytorch_with_examples.html
https://pytorch.org/tutorials/beginner/former_torchies/nnft_tutorial.html
PyTorch公式でチートシートを用意してくれているので、そこを見ると習得が速いかも。
チートシートの内容を画像化して公開している人もいる↓
小技集をJupyter Notebookにまとめている方もいる↓
https://github.com/bfortuner/pytorch-cheatsheet/blob/master/pytorch-cheatsheet.ipynb
あとは、torchvisionに有名な畳み込みニューラルネットワークが実装されているので、それを読むと真似しやすいかも↓
https://pytorch.org/docs/stable/torchvision/models.html
追記:PyTorch専用IDEも登場してる↓
関連記事
AndroidもopenGLも初心者さ (でもJavaは知っ...
SONY製のニューラルネットワークライブラリ『NNabla』
ドットインストールのWordPress入門レッスン
Kornia:微分可能なコンピュータービジョンライブラリ
OpenFace:Deep Neural Networkによ...
OpenCVで平均顔を作るチュートリアル
iPhone・iPod touchで動作する知育ロボット『R...
UnityでOpenCVを使うには?
CNN Explainer:畳み込みニューラルネットワーク可...
オープンソースの人体モデリングツール『MakeHuman』の...
書籍『3次元コンピュータビジョン計算ハンドブック』を購入
WebGL開発に関する情報が充実してきている
続・ディープラーニングの資料
Konashiを買った
GoB:ZBrushとBlenderを連携させるアドオン
Google App Engine上のWordPressでF...
C++の抽象クラス
Deep Learningとその他の機械学習手法の性能比較
Unityからkonashiをコントロールする
書籍『ROSプログラミング』
このブログのデザインに飽きてきた
html5のcanvasの可能性
FreeMoCap Project:オープンソースのマーカー...
SVM (Support Vector Machine)
3分の動画でプログラミングを学習できるサイト『ドットインスト...
法線マップを用意してCanvas上でShadingするサンプ...
DensePose:画像中の人物表面のUV座標を推定する
Quartus II
WordPressのテーマを自作する
ROMOハッカソンに行ってきた
ZBrush 2018での作業環境を整える
DCGAN (Deep Convolutional GAN)...
LLM Visualization:大規模言語モデルの可視化
ポートレート撮影のためのスタジオ照明の基本
CGのためのディープラーニング
Unityで強化学習できる『Unity ML-Agents』
Faster R-CNN:ディープラーニングによる一般物体検...
Deep Fluids:流体シミュレーションをディープラーニ...
Python2とPython3
Live CV:インタラクティブにComputer Visi...
GeoGebra:無料で使える数学アプリ
Amazon EC2ログイン用の秘密鍵を無くした場合の対処方...
コメント