Skip to content

Commit

Permalink
device usage example
Browse files Browse the repository at this point in the history
it is not explained how to select a specific GPU, as neither of
`--device 1`
`--device nvidia1`
`--device /dev/nvidia1`
work.

`--device cuda:<n>` does, but is not intuitive so I suggest this example addition.
  • Loading branch information
FraYoshi authored Jan 20, 2025
1 parent 517a43e commit 19c785e
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -93,6 +93,10 @@ Adding `--task translate` will translate the speech into English:

whisper japanese.wav --language Japanese --task translate

Adding `--device cuda:<n>` will use the selected cuda GPU number

whisper japanese.wav --language Japanese --task translate --device cuda:1

Run the following to view all available options:

whisper --help
Expand Down

0 comments on commit 19c785e

Please # to comment.