Add not about GPU requirements
This commit is contained in:
@@ -16,6 +16,12 @@ The model conversion requires the modules `transformers` and `torch` which are i
|
||||
pip install -e .
|
||||
```
|
||||
|
||||
### GPU support
|
||||
|
||||
GPU execution requires the NVIDIA libraries cuBLAS 11.x and cuDNN 8.x to be installed on the system. Please refer to the [CTranslate2 documentation](https://opennmt.net/CTranslate2/installation.html).
|
||||
|
||||
**Note for Windows users:** the published Windows wheels are currently not built with cuDNN, which is required to run the Whisper model. Consequently GPU execution is currently not possible on Windows out of the box. If you feel adventurous, a possible workaround is to [compile from sources](https://opennmt.net/CTranslate2/installation.html#install-from-sources).
|
||||
|
||||
## Usage
|
||||
|
||||
### Model conversion
|
||||
|
||||
Reference in New Issue
Block a user