Skip to content

Commit

Permalink
doc patch
Browse files Browse the repository at this point in the history
  • Loading branch information
luiskuhn committed Feb 19, 2024
1 parent 181ef14 commit 2ac8854
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions docs/usage.rst
Original file line number Diff line number Diff line change
Expand Up @@ -49,10 +49,10 @@ Training using GPUs

Regularly used commands for development and testing:

``mlflow run . --build-image -A runtime=nvidia``
``mlflow run . --build-image -A runtime=nvidia -P max_epochs=3``
``mlflow run . --build-image -A runtime=nvidia -P max_epochs=3 -P gpus=1``
``mlflow run . --build-image -A runtime=nvidia -P max_epochs=3 -P gpus=2``
- ``mlflow run . --build-image -A runtime=nvidia``
- ``mlflow run . --build-image -A runtime=nvidia -P max_epochs=3``
- ``mlflow run . --build-image -A runtime=nvidia -P max_epochs=3 -P gpus=1``
- ``mlflow run . --build-image -A runtime=nvidia -P max_epochs=3 -P gpus=2``

Conda environments will automatically use the GPU if available. Docker requires the accessible GPUs to be passed as runtime parameters. To train using all gpus run ``mlflow run . --build-image -A runtime=nvidia -P gpus=<<num_of_gpus>> -P acc=ddp``.

Expand Down

0 comments on commit 2ac8854

Please sign in to comment.