explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-11-10 19:57:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update cupy extras (#11279)

Browse Source 
* Update cupy extras:

* Extend to v11
* Add `cupy-cuda11x` and `cupy-wheel`
* Update quickstart to use `cupy-wheel` for CUDA 10.2+

* Rename cuda-wheel to cuda-autodetect, remove repeated CUDA in menu
This commit is contained in:
Adriane Boyd 2022-09-13 09:04:53 +02:00 committed by GitHub
parent 0ec9a696e6
commit 6be6913ba5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 22 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
setup.cfg
View File

@ -76,37 +76,41 @@ transformers =
ray = ray =
spacy_ray>=0.1.0,<1.0.0 spacy_ray>=0.1.0,<1.0.0
cuda = cuda =
cupy>=5.0.0b4,<11.0.0 cupy>=5.0.0b4,<12.0.0
cuda80 = cuda80 =
cupy-cuda80>=5.0.0b4,<11.0.0 cupy-cuda80>=5.0.0b4,<12.0.0
cuda90 = cuda90 =
cupy-cuda90>=5.0.0b4,<11.0.0 cupy-cuda90>=5.0.0b4,<12.0.0
cuda91 = cuda91 =
cupy-cuda91>=5.0.0b4,<11.0.0 cupy-cuda91>=5.0.0b4,<12.0.0
cuda92 = cuda92 =
cupy-cuda92>=5.0.0b4,<11.0.0 cupy-cuda92>=5.0.0b4,<12.0.0
cuda100 = cuda100 =
cupy-cuda100>=5.0.0b4,<11.0.0 cupy-cuda100>=5.0.0b4,<12.0.0
cuda101 = cuda101 =
cupy-cuda101>=5.0.0b4,<11.0.0 cupy-cuda101>=5.0.0b4,<12.0.0
cuda102 = cuda102 =
cupy-cuda102>=5.0.0b4,<11.0.0 cupy-cuda102>=5.0.0b4,<12.0.0
cuda110 = cuda110 =
cupy-cuda110>=5.0.0b4,<11.0.0 cupy-cuda110>=5.0.0b4,<12.0.0
cuda111 = cuda111 =
cupy-cuda111>=5.0.0b4,<11.0.0 cupy-cuda111>=5.0.0b4,<12.0.0
cuda112 = cuda112 =
cupy-cuda112>=5.0.0b4,<11.0.0 cupy-cuda112>=5.0.0b4,<12.0.0
cuda113 = cuda113 =
cupy-cuda113>=5.0.0b4,<11.0.0 cupy-cuda113>=5.0.0b4,<12.0.0
cuda114 = cuda114 =
cupy-cuda114>=5.0.0b4,<11.0.0 cupy-cuda114>=5.0.0b4,<12.0.0
cuda115 = cuda115 =
cupy-cuda115>=5.0.0b4,<11.0.0 cupy-cuda115>=5.0.0b4,<12.0.0
cuda116 = cuda116 =
cupy-cuda116>=5.0.0b4,<11.0.0 cupy-cuda116>=5.0.0b4,<12.0.0
cuda117 = cuda117 =
cupy-cuda117>=5.0.0b4,<11.0.0 cupy-cuda117>=5.0.0b4,<12.0.0
cuda11x =
cupy-cuda11x>=11.0.0,<12.0.0
cuda-autodetect =
cupy-wheel>=11.0.0,<12.0.0
apple = apple =
thinc-apple-ops>=0.1.0.dev0,<1.0.0 thinc-apple-ops>=0.1.0.dev0,<1.0.0
# Language tokenizers with external dependencies # Language tokenizers with external dependencies

12
website/src/widgets/quickstart-install.js
View File

@ -9,7 +9,7 @@ const DEFAULT_PLATFORM = 'x86'
const DEFAULT_MODELS = ['en'] const DEFAULT_MODELS = ['en']
const DEFAULT_OPT = 'efficiency' const DEFAULT_OPT = 'efficiency'
const DEFAULT_HARDWARE = 'cpu' const DEFAULT_HARDWARE = 'cpu'
const DEFAULT_CUDA = 'cuda113' const DEFAULT_CUDA = 'cuda-autodetect'
const CUDA = { const CUDA = {
'8.0': 'cuda80', '8.0': 'cuda80',
'9.0': 'cuda90', '9.0': 'cuda90',
@ -17,15 +17,7 @@ const CUDA = {
'9.2': 'cuda92', '9.2': 'cuda92',
'10.0': 'cuda100', '10.0': 'cuda100',
'10.1': 'cuda101', '10.1': 'cuda101',
'10.2': 'cuda102', '10.2, 11.0+': 'cuda-autodetect',
'11.0': 'cuda110',
'11.1': 'cuda111',
'11.2': 'cuda112',
'11.3': 'cuda113',
'11.4': 'cuda114',
'11.5': 'cuda115',
'11.6': 'cuda116',
'11.7': 'cuda117',
} }
const LANG_EXTRAS = ['ja'] // only for languages with models const LANG_EXTRAS = ['ja'] // only for languages with models