From af07fc3bc1abc07a00f7860addbdabf4523ad2bf Mon Sep 17 00:00:00 2001 From: Santiago Castro Date: Tue, 30 Mar 2021 03:47:33 -0400 Subject: [PATCH] Add support for CUDA 11.2 (#7583) * Add support for CUDA 11.2 * Update the docs * Format Co-authored-by: Adriane Boyd --- setup.cfg | 2 ++ website/docs/usage/index.md | 6 +++--- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/setup.cfg b/setup.cfg index 09f989c54..e928e90a6 100644 --- a/setup.cfg +++ b/setup.cfg @@ -91,6 +91,8 @@ cuda110 = cupy-cuda110>=5.0.0b4,<9.0.0 cuda111 = cupy-cuda111>=5.0.0b4,<9.0.0 +cuda112 = + cupy-cuda112>=5.0.0b4,<9.0.0 # Language tokenizers with external dependencies ja = sudachipy>=0.4.9 diff --git a/website/docs/usage/index.md b/website/docs/usage/index.md index cbbda2e4f..665d334f8 100644 --- a/website/docs/usage/index.md +++ b/website/docs/usage/index.md @@ -130,9 +130,9 @@ which provides a numpy-compatible interface for GPU arrays. spaCy can be installed on GPU by specifying `spacy[cuda]`, `spacy[cuda90]`, `spacy[cuda91]`, `spacy[cuda92]`, `spacy[cuda100]`, `spacy[cuda101]`, -`spacy[cuda102]`, `spacy[cuda110]` or `spacy[cuda111]`. If you know your cuda -version, using the more explicit specifier allows cupy to be installed via -wheel, saving some compilation time. The specifiers should install +`spacy[cuda102]`, `spacy[cuda110]`, `spacy[cuda111]` or `spacy[cuda112]`. If you +know your cuda version, using the more explicit specifier allows cupy to be +installed via wheel, saving some compilation time. The specifiers should install [`cupy`](https://cupy.chainer.org). ```bash