Add not about GPU requirements

This commit is contained in:
Guillaume Klein
2023-02-12 12:04:11 +01:00
parent 60e667e0d2
commit 3e7b8109cd

View File

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