差分
このページの2つのバージョン間の差分を表示します。
| 両方とも前のリビジョン 前のリビジョン 次のリビジョン | 前のリビジョン | ||
| mediapipe:モデル学習 [2023/09/19 14:52] – baba | mediapipe:モデル学習 [2023/10/02 15:14] (現在) – [tensorlow on gpu] baba | ||
|---|---|---|---|
| 行 32: | 行 32: | ||
| conda create -n mediapipe python=3.9 | conda create -n mediapipe python=3.9 | ||
| - | conda activate | + | conda activate |
| conda install -c conda-forge cudatoolkit=11.8.0 | conda install -c conda-forge cudatoolkit=11.8.0 | ||
| python3 -m pip install nvidia-cudnn-cu11==8.6.0.163 tensorflow==2.12.* | python3 -m pip install nvidia-cudnn-cu11==8.6.0.163 tensorflow==2.12.* | ||
| 行 48: | 行 48: | ||
| * https:// | * https:// | ||
| + | ===== Multiple GPU ===== | ||
| GPUを利用刷る場合、上記で設定がうまく行っていればそのままGPUで学習が始まる。一方で複数GPUを持っている場合は、別途オプションを設定する必要がある。 | GPUを利用刷る場合、上記で設定がうまく行っていればそのままGPUで学習が始まる。一方で複数GPUを持っている場合は、別途オプションを設定する必要がある。 | ||
| https:// | https:// | ||
| 行 54: | 行 55: | ||
| 等と設定することで、複数gpuを学習に利用できる。ただし、実際にこのオプションを設定して実行すると distribution_strageryが' | 等と設定することで、複数gpuを学習に利用できる。ただし、実際にこのオプションを設定して実行すると distribution_strageryが' | ||
| * num_gpus:4, | * num_gpus:4, | ||
| - | * distribution_strategy:' | + | * distribution_strategy=' |
| - | でOK. | + | でOK.例えばこんな感じ |
| + | < | ||
| + | hparams = object_detector.HParams( | ||
| + | export_dir=' | ||
| + | epochs=120, | ||
| + | num_gpus=4, | ||
| + | distribution_strategy=' | ||
| + | ) | ||
| + | </ | ||
| ====== 実行 ====== | ====== 実行 ====== | ||
| うまくいくとモデルが完成するので，出来上がったモデルを試すページは以下 | うまくいくとモデルが完成するので，出来上がったモデルを試すページは以下 | ||