差分
このページの2つのバージョン間の差分を表示します。
両方とも前のリビジョン 前のリビジョン 次のリビジョン | 前のリビジョン | ||
mediapipe:モデル学習 [2023/09/30 22:28] – [Multiple GPU] 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.* | ||
行 55: | 行 55: | ||
等と設定することで、複数gpuを学習に利用できる。ただし、実際にこのオプションを設定して実行すると distribution_strageryが' | 等と設定することで、複数gpuを学習に利用できる。ただし、実際にこのオプションを設定して実行すると distribution_strageryが' | ||
* num_gpus:4, | * num_gpus:4, | ||
- | * distribution_strategy:' | + | * distribution_strategy=' |
- | でOK. | + | でOK.例えばこんな感じ |
< | < |