このページの2つのバージョン間の差分を表示します。
両方とも前のリビジョン 前のリビジョン | 次のリビジョン両方とも次のリビジョン |
nvidia-docker:caffe:mobilenetssd [2020/03/02 02:42] – [SSD用caffeリポジトリのクローン] baba | nvidia-docker:caffe:mobilenetssd [2020/03/02 10:32] – [Dockerfileの修正] baba |
---|
| |
===== 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の修正箇所のみ記述しています. |