このページの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の修正箇所のみ記述しています． |