From 3e7b8109cdf9d5f4b379dc91cf5246331d571866 Mon Sep 17 00:00:00 2001 From: Guillaume Klein Date: Sun, 12 Feb 2023 12:04:11 +0100 Subject: [PATCH] Add not about GPU requirements --- README.md | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/README.md b/README.md index fb6bcc2..0bf2c62 100644 --- a/README.md +++ b/README.md @@ -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