explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-01-13 18:56:36 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add support for CUDA 11.2 (#7583)

Browse Source 
* Add support for CUDA 11.2

* Update the docs

* Format

Co-authored-by: Adriane Boyd <adrianeboyd@gmail.com>
This commit is contained in:
Santiago Castro 2021-03-30 03:47:33 -04:00 committed by GitHub
parent 5b4dde38a3
commit af07fc3bc1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
setup.cfg
View File

@ -91,6 +91,8 @@ cuda110 =
cupy-cuda110>=5.0.0b4,<9.0.0 cupy-cuda110>=5.0.0b4,<9.0.0
cuda111 = cuda111 =
cupy-cuda111>=5.0.0b4,<9.0.0 cupy-cuda111>=5.0.0b4,<9.0.0
cuda112 =
cupy-cuda112>=5.0.0b4,<9.0.0
# Language tokenizers with external dependencies # Language tokenizers with external dependencies
ja = ja =
sudachipy>=0.4.9 sudachipy>=0.4.9

6
website/docs/usage/index.md
View File

@ -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 can be installed on GPU by specifying `spacy[cuda]`, `spacy[cuda90]`,
`spacy[cuda91]`, `spacy[cuda92]`, `spacy[cuda100]`, `spacy[cuda101]`, `spacy[cuda91]`, `spacy[cuda92]`, `spacy[cuda100]`, `spacy[cuda101]`,
`spacy[cuda102]`, `spacy[cuda110]` or `spacy[cuda111]`. If you know your cuda `spacy[cuda102]`, `spacy[cuda110]`, `spacy[cuda111]` or `spacy[cuda112]`. If you
version, using the more explicit specifier allows cupy to be installed via know your cuda version, using the more explicit specifier allows cupy to be
wheel, saving some compilation time. The specifiers should install installed via wheel, saving some compilation time. The specifiers should install
[`cupy`](https://cupy.chainer.org). [`cupy`](https://cupy.chainer.org).
```bash ```bash