Skip to content

Commit

Permalink
Adding note CUDA / GPU support for Windows (#810)
Browse files Browse the repository at this point in the history
  • Loading branch information
raivisdejus committed Jun 22, 2024
1 parent 2e7dbab commit 9894760
Showing 1 changed file with 13 additions and 1 deletion.
14 changes: 13 additions & 1 deletion CONTRIBUTING.md
Original file line number Diff line number Diff line change
Expand Up @@ -88,4 +88,16 @@ cp .\dll_backup\whisper.dll .\buzz\
$whlFile = Get-ChildItem .\dist\buzz*.whl | Select-Object -First 1
pip install $whlFile
```
13. Run Buzz `python -m buzz`
13. Run Buzz `python -m buzz`

#### GPU Support

To enable GPU support first ensure CUDA 12.1 is installed - https://developer.nvidia.com/cuda-12-1-0-download-archive

Switch torch library to GPU version
```
pip3 unstall torch torchaudio
pip3 install torch==2.2.1+cu121 torchaudio==2.2.1+cu121 --index-url https://download.pytorch.org/whl/cu121
```

Run Buzz `python -m buzz`

0 comments on commit 9894760

Please sign in to comment.