Adding note CUDA / GPU support for Windows (#810)

This commit is contained in:
Raivis Dejus 2024-06-22 09:48:57 +03:00 committed by GitHub
commit 98947602e4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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`