このページの2つのバージョン間の差分を表示します。
次のリビジョン | 前のリビジョン 次のリビジョン両方とも次のリビジョン |
nvidia-docker:caffe:mobilenetssd [2020/03/01 23:26] – 作成 baba | nvidia-docker:caffe:mobilenetssd [2020/03/02 10:32] – [Dockerfileの修正] baba |
---|
====== MobielNetSSDをnvidia-dockerで環境構築して学習されるまで ====== | ====== MobielNetSSDをnvidia-dockerで環境構築 ====== |
* https://github.com/chuanqi305/MobileNet-SSD | * https://github.com/chuanqi305/MobileNet-SSD |
の環境を作る際,READMEに従う際にSSD用のcaffe環境を別gitからcloneした後にMobileNet-SSDの学習に | の環境を作る際,READMEに従う際にSSD用のcaffe環境を別gitからcloneした後にMobileNet-SSDの学習に |
===== SSD用caffeリポジトリのクローン ===== | ===== SSD用caffeリポジトリのクローン ===== |
まず,https://github.com/chuanqi305/MobileNet-SSD に記載されているように,BVLCのcaffeではなく,著者のweiliu89さんのcaffeを最初にクローンします.ただほしいのはここのdockerディレクトリ以下だけなので他は使いません. | まず,https://github.com/chuanqi305/MobileNet-SSD に記載されているように,BVLCのcaffeではなく,著者のweiliu89さんのcaffeを最初にクローンします.ただほしいのはここのdockerディレクトリ以下だけなので他は使いません. |
$ git clone https://github.com/weiliu89/caffe.git | git clone https://github.com/weiliu89/caffe.git |
$ cd caffe/docker | cd caffe/docker |
| |
===== Dockerfileの修正 ===== | ===== Dockerfileの修正 ===== |
んでもって,このdockerにあるREADME( https://github.com/weiliu89/caffe/tree/ssd/docker )をみると,すぐにdockerでビルドする段取りが記述されています.これをnvidia-dockerでgpuに置き換えるわけですが,このビルドを実行する前に standalone/gpu/Dockerfile を修正します.なぜならこのDockerfileはBVLCそのままでssd用のDockerfileになっていないからです.修正する箇所は git clone を行っているところです.BVLCのcaffeではなく,weiliu89のcaffeにすること,さらにENV CLONE_TAGをmaster からssdに修正します. | んでもって,このdockerにあるREADME( https://github.com/weiliu89/caffe/tree/ssd/docker )をみると,すぐにdockerでビルドする段取りが記述されています.これをnvidia-dockerでgpuに置き換えるわけですが,このビルドを実行する前に standalone/gpu/Dockerfile を修正します.なぜならこのDockerfileはBVLCそのままでssd用のDockerfileになっていないからです.修正する箇所は git clone を行っているところです.BVLCのcaffeではなく,weiliu89のcaffeにすること,さらにENV CLONE_TAGをmaster からssdに修正します.この修正を行わずにビルドするとBVLCのcaffeでmasterが入ってしまい,caffeのコンパイル自体は成功するがMobileNet-SSDの学習の際,必要な実装部分が足りずにプロセスが死んでしまいます.https://github.com/chuanqi305/MobileNet-SSD/issues/157 に記載されているように困った人もいたようです.提供されてるDockerfile使わなかければこんなことは起こりません. |
<file> | <file> |
# standalone/gpu/Dockerfileの修正箇所のみ記述しています. | # standalone/gpu/Dockerfileの修正箇所のみ記述しています. |